Spec-Zone.ru › TensorFlow 1.15

tf.IndexedSlices

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

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

Псевдонимы

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

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.IndexedSlices, `tf.compat.v2.IndexedSlices`

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API