tf.tensor_scatter_nd_add

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

tf.tensor_scatter_nd_add(
    tensor: Annotated[Any, TV_TensorScatterAdd_T],
    indices: Annotated[Any, TV_TensorScatterAdd_Tindices],
    updates: Annotated[Any, TV_TensorScatterAdd_T],
    name=None
) -> Annotated[Any, TV_TensorScatterAdd_T]

Используется в блокнотах

Эта операция создает новый тензор, добавляя разреженные updates к переданному tensor. Эта операция очень похожа на tf.compat.v1.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_nd_add — добавлять отдельные элементы в тензор по индексу. Например, предположим, что мы хотим добавить 4 элемента в тензор ранга 1, содержащий 8 элементов.

В Python операция добавления с использованием scatter выглядела бы так:

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)
updated
<tf.Tensor: shape=(8,), dtype=int32,
numpy=array([ 1, 12,  1, 11, 10,  1,  1, 13], dtype=int32)>

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

В Python операция добавления с использованием scatter выглядела бы так:

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)
updated
<tf.Tensor: shape=(4, 4, 4), dtype=int32,
numpy=array([[[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]]], dtype=int32)>

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