tf.tensor_scatter_nd_add

Добавляет разреженные updates к существующему тензору в соответствии с indices.

tf.tensor_scatter_nd_add(
    tensor, indices, updates, name=None
)

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

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

indices.shape[-1] <= tensor.shape.rank

Последнее измерение indices соответствует индексам элементов (если indices.shape[-1] = tensor.shape.rank) или срезов (если indices.shape[-1] < tensor.shape.rank) по размеру indices.shape[-1] тензора tensor.shape. updates — это тензор с формой

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

Простейшая форма операции tensor_scatter_add заключается в добавлении отдельных элементов в тензор по индексу. Например, предположим, что мы хотим добавить 4 элемента в тензор ранга 1 с 8 элементами.

В Python операция scatter add будет выглядеть следующим образом:

indices = tf.constant([[4], [3], [1], [7]])
updates = tf.constant([9, 10, 11, 12])
tensor = tf.ones([8], dtype=tf.int32)
updated = tf.tensor_scatter_nd_add(tensor, indices, updates)
print(updated)

Результирующий тензор будет выглядеть следующим образом:

[1, 12, 1, 11, 10, 1, 1, 13]

Мы также можем вставить целые срезы тензора более высокого ранга сразу. Например, если мы хотели вставить два среза в первое измерение тензора ранга 3 с двумя матрицами новых значений.

В Python операция scatter add будет выглядеть следующим образом:

indices = tf.constant([[0], [2]])
updates = tf.constant([[[5, 5, 5, 5], [6, 6, 6, 6],
                        [7, 7, 7, 7], [8, 8, 8, 8]],
                       [[5, 5, 5, 5], [6, 6, 6, 6],
                        [7, 7, 7, 7], [8, 8, 8, 8]]])
tensor = tf.ones([4, 4, 4],dtype=tf.int32)
updated = tf.tensor_scatter_nd_add(tensor, indices, updates)
print(updated)

Результирующий тензор будет выглядеть следующим образом:

[[[6, 6, 6, 6], [7, 7, 7, 7], [8, 8, 8, 8], [9, 9, 9, 9]],
 [[1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1]],
 [[6, 6, 6, 6], [7, 7, 7, 7], [8, 8, 8, 8], [9, 9, 9, 9]],
 [[1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1]]]

Обратите внимание, что на CPU, если найден индекс за пределами границ, возвращается ошибка. На GPU, если найден индекс за пределами границ, индекс игнорируется.