tf.raw_ops.TensorScatterSub

Вычитает разреженные updates из существующего тензора в соответствии с indices.

tf.raw_ops.TensorScatterSub(
    tensor, indices, updates, name=None
)

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

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

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

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

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

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

В Python эта операция разреженного вычитания выглядит так:

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_sub(tensor, indices, updates)
print(updated)

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

[1, -10, 1, -9, -8, 1, 1, -11]

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

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

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_sub(tensor, indices, updates)
print(updated)

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

[[[-4, -4, -4, -4], [-5, -5, -5, -5], [-6, -6, -6, -6], [-7, -7, -7, -7]],
 [[1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1]],
 [[-4, -4, -4, -4], [-5, -5, -5, -5], [-6, -6, -6, -6], [-7, -7, -7, -7]],
 [[1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1]]]

Обратите внимание: на ЦПУ при обнаружении индекса за пределами допустимого диапазона возвращается ошибка. На ГПУ такой индекс игнорируется.