tf.raw_ops.TensorScatterUpdate
Рассеять updates в существующий тензор в соответствии с indices.
tf.raw_ops.TensorScatterUpdate(
tensor, indices, updates, name=None
)
Эта операция создает новый тензор, применяя разреженные updates к переданному tensor. Эта операция очень похожа на tf.scatter_nd, за исключением того, что обновления применяются к существующему тензору (а не к тензору из нулей). Если память существующего тензора нельзя использовать повторно, создается его копия, которая затем обновляется.
Если indices содержит дубликаты, для индекса выбирается последнее обновление.
Если на CPU обнаружен индекс, выходящий за границы, возвращается ошибка.
- Если обнаружен индекс, выходящий за границы, он игнорируется.
- Порядок применения обновлений не детерминирован, поэтому результат также будет недетерминированным, если
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