tf.IndexedSlices

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

Просмотреть псевдонимы

Псевдонимы для совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.IndexedSlices

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

Используется в блокнотах

Используется в учебных материалах
Федеративное обучение с эффективным использованием ресурсов клиента для больших моделей с помощью `federated_select` и разреженной агрегации

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

values: Tensor любого типа данных с формой [D0, D1, ..., Dn].
indices: целочисленный Tensor размерности 1 с формой [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, в котором используются многомерные индексы и скалярные значения.

Атрибуты
`dense_shape`	Одномерный `Tensor`, содержащий форму соответствующего плотного тензора.
`device`	Имя устройства, на котором будет создан `values`, или `None`.
`dtype`	`DType` элементов в этом тензоре.
`graph`	`Graph`, содержащий тензоры значений, индексов и формы.
`indices`	Одномерный `Tensor`, содержащий индексы срезов.
`name`	Имя этого `IndexedSlices`.
`op`	`Operation`, который выдаёт `values` в качестве результата.
`shape`	Возвращает `tf.TensorShape`, представляющий форму плотного тензора.
`values`	`Tensor`, содержащий значения срезов.

Методы

`consumers`

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

consumers()

`neg`

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

__neg__()

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/IndexedSlices