TensorScatterAdd

Добавляет редкие «обновления» к существующему тензору в соответствии с «индексами».

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

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

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

Последнее измерение индексов соответствует индексам элементов (если 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 эта операция добавления разброса будет выглядеть так:

>>> индексы = tf.constant([[4], [3], [1], [7]]) >>> обновления = tf.constant([9, 10, 11, 12]) >>> тензор = tf.ones([8], dtype=tf.int32) >>> обновлено = tf.tensor_scatter_nd_add(тензор, индексы, обновления) >>> обновлено

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

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

>>> индексы = tf.constant([[0], [2]]) >>> обновления = 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) >>> обновлено = tf.tensor_scatter_nd_add(тензор, индексы, обновления) >>> обновлено

Примечание. Если на ЦП обнаружен выходной индекс, возвращается ошибка. На графическом процессоре, если обнаружен выходящий за пределы индекс, индекс игнорируется.