tf.IndexedSlices
| Просмотреть исходный код на GitHub |
Разреженное представление набора тензорных слайсов по заданным индексам.
tf.IndexedSlices(
values, indices, dense_shape=None
)
Этот класс является простым оболочкой для пары Tensor объектов:
-
values:Tensorлюбого типа с формой[D0, D1, ..., Dn]. -
indices: ЦелочисленныйTensor1-мерный размерностью[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.4/api_docs/python/tf/IndexedSlices