tf.IndexedSlices

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

Псевдонимы

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

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.IndexedSlices

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

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

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

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

values: Tensor любого типа с формой [D0, D1, ..., Dn].
indices: 1-мерный целочисленный Tensor с формой [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`	1-мерный `Tensor`, содержащий форму соответствующего плотного тензора.
`device`	Имя устройства, на котором будет создан `values`, или `None`.
`dtype`	Тип элементов в этом тензоре.
`graph`	`Graph`, содержащий тензоры значений, индексов и формы.
`indices`	1-мерный `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