阅读(9.4k) 书签 (0)

TensorFlow函数:IndexedSlices

2018-10-26 17:33 更新
函数:tf.IndexedSlices

IndexedSlices 类

定义在:tensorflow/python/framework/ops.py.

参见指南:变量>稀疏变量更新

在给定索引上的一组张量切片的稀疏表示.
此类是一对张量对象的简单包装:

  • values:具有形状 [D0, D1, ..., Dn] 的任何类型的张量.
  • indices:具有形状 [D0] 的一维整数张量.

IndexedSlices 通常用于表示较大张量形状为 [LARGE0, D1,.., DN] 的子集,其中 LARGE0 >> D0.指数中的值是从较大张量中提取的切片的第一个维度中的索引.

由 IndexedSlices 切片表示的稠密张量有:

dense[slices.indices[i], :, :, :, ...] = slices.values[i, :, :, :, ...]

IndexedSlices 类主要用于具有稀疏渐变 (如 tf.gather) 的操作的渐变定义.

将此表示与 tf.SparseTensor 进行对比,它使用多维指数和标量值.

属性

  • dense_shape
    包含相应稠密张量的形状的1维张量.
  • device
    将在其上生成值的设备的名称,或 None.
  • dtype
    此张量中元素的 DType.
  • graph
    包含值、索引和形状张量的图.
  • indices
    包含切片的索引的1维张量.
  • name
    该 IndexedSlices 的名字.
  • op
    将值作为输出生成的操作.
  • values
    包含切片值的张量.

方法

__init__

__init__(
    values,
    indices,
    dense_shape=None
)

创建一个 IndexedSlices.

__neg__

__neg__()