Spec-Zone.ru › TensorFlow

tf.raw_ops.TensorScatterUpdate

Рассеять updates в существующий тензор в соответствии с indices.

Просмотреть псевдонимы

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

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

tf.compat.v1.raw_ops.TensorScatterUpdate

tf.raw_ops.TensorScatterUpdate(
    tensor, indices, updates, name=None
)

Эта операция создает новый тензор, применяя разреженные updates к переданному tensor. Эта операция очень похожа на tf.scatter_nd, за исключением того, что обновления применяются к существующему тензору (а не к тензору из нулей). Если память существующего тензора нельзя использовать повторно, создается его копия, которая затем обновляется.

Если indices содержит дубликаты, для индекса выбирается последнее обновление.

Если на CPU обнаружен индекс, выходящий за границы, возвращается ошибка.

Предупреждение: Для этой операции предусмотрена особая семантика работы на GPU.
  • Если обнаружен индекс, выходящий за границы, он игнорируется.
  • Порядок применения обновлений не детерминирован, поэтому результат также будет недетерминированным, если indices содержит дубликаты.

indices — это целочисленный тензор, содержащий индексы в новом тензоре формы shape.

  • indices должен иметь не менее двух осей: (num_updates, index_depth).
  • Последняя ось indices определяет глубину индексации в tensor, поэтому эта глубина индексации должна быть меньше ранга tensor: indices.shape[-1] <= tensor.ndim

если indices.shape[-1] = tensor.rank, этот Op индексирует и обновляет скалярные элементы. если indices.shape[-1] < tensor.rank, он индексирует и обновляет срезы входного tensor.

Ранг каждого update равен tensor.rank - indices.shape[-1]. Общая форма updates:

indices.shape[:-1] + tensor.shape[indices.shape[-1]:]

Примеры использования см. в функции Python tf.tensor_scatter_nd_update

Аргументы
tensor Тензор типа Tensor. Тензор для копирования или обновления.
indices Тензор типа Tensor. Должен иметь один из следующих типов: int16, int32, int64, uint16. Тензор индексов.
updates Тензор типа Tensor. Должен иметь тот же тип, что и tensor. Обновления для рассеивания в выходной тензор.
name Имя операции (необязательно).
Возвращает
Тензор типа Tensor. Имеет тот же тип, что и tensor.

© 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/raw_ops/TensorScatterUpdate

Spec-Zone.ru

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