Spec-Zone.ru › TensorFlow 2.9

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 содержит дубликаты, мы выбираем последнее обновление для индекса.

Если на процессоре найден индекс за пределами границ, возвращается ошибка.

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

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

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

Если indices.shape[-1] = tensor.rank, эта операция индексирует и обновляет скалярные элементы. Если 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 A Tensor. Тензор для копирования/обновления.
indices A Tensor. Должен быть одного из следующих типов: int16, int32, int64, uint16. Тензор индексов.
updates A Tensor. Должен иметь тот же тип, что и tensor. Обновления для разброса в выходные данные.
name Имя операции (необязательно).
Возвращает
A 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/versions/r2.9/api_docs/python/tf/raw_ops/TensorScatterUpdate

Spec-Zone.ru

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