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.SparseTensor, которая использует многомерные индексы и скалярные значения.
| Атрибуты | |
|---|---|
dense_shape | Одномерный Tensor, содержащий форму соответствующего плотного тензора. |
device | Имя устройства, на котором будет произведен values, или None. |
dtype | Тип элементов в этом тензоре. |
graph | Тензор, содержащий значения, индексы и форму тензоров. |
indices | Одномерный Tensor, содержащий индексы слайсов. |
name | Имя этого IndexedSlices. |
op | Тензор, который производит values в качестве результата. |
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/r1.15/api_docs/python/tf/IndexedSlices