tf.IndexedSlices

Разреженное представление набора тензорных слайсов по заданным индексам.

tf.IndexedSlices(
    values, indices, dense_shape=None
)

Этот класс — простой оболочкой для пары Tensor объектов:

• values: Tensor произвольного типа с формой [D0, D1, ..., Dn].
• indices: Целочисленный Tensor 1-D с формой [D0].

IndexedSlices обычно используется для представления подмножества большего тензора dense с формой [LARGE0, D1, .. , DN], где LARGE0 >> D0. Значения в indices — индексы в первом измерении слайсов, которые были извлечены из большего тензора.

Плотный тензор dense, представленный IndexedSlices slices, имеет

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

Класс IndexedSlices в основном используется в определении градиентов для операций с разреженными градиентами (например, tf.gather).

v = tf.Variable([[0.,1, 2], [2, 3, 4], [4, 5, 6], [6, 7, 8]])
with tf.GradientTape() as tape:
  r = tf.gather(v, [1,3])
index_slices = tape.gradient(r,v)
index_slices
<...IndexedSlices object ...>
index_slices.indices.numpy()
array([1, 3], dtype=int32)
index_slices.values.numpy()
array([[1., 1., 1.],
       [1., 1., 1.]], dtype=float32)

Сравните это представление с tf.sparse.SparseTensor, которое использует многомерные индексы и скалярные значения.

Методы

consumers

Просмотреть исходный код

consumers()

__neg__

Просмотреть исходный код

__neg__()