tf.IndexedSlices
| Просмотреть исходный код на GitHub |
Распарченное представление набора тензорных слайсов в заданных индексах.
tf.IndexedSlices(
values, indices, dense_shape=None
)
Этот класс представляет собой простой оболочку для пары 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).
Сравните это представление с 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__()
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/IndexedSlices