torch.Tensor.scatter_reduce_

Tensor.scatter_reduce_(dim, index, src, reduce, *, include_self=True) → Tensor

Сводит все значения из тензора src к индексам, указанным в тензоре index в тензоре self с использованием заданного сворачивания, определённого аргументом reduce ("sum", "prod", "mean", "amax", "amin"). Для каждого значения в src, оно сводится к индексу в self, который определяется его индексом в src для dimension != dim и соответствующим значением в index для dimension = dim. Если include_self="True", значения в тензоре self включаются в сворачивание.

self, index и src должны иметь одинаковое количество измерений. Также требуется, чтобы index.size(d) <= src.size(d) для всех измерений d, и чтобы index.size(d) <= self.size(d) для всех измерений d != dim. Обратите внимание, что index и src не транслируются.

Для 3-мерного тензора с reduce="sum" и include_self=True результат представлен следующим образом:

self[index[i][j][k]][j][k] += src[i][j][k]  # if dim == 0
self[i][index[i][j][k]][k] += src[i][j][k]  # if dim == 1
self[i][j][index[i][j][k]] += src[i][j][k]  # if dim == 2

Примечание

Эта операция может иметь недетерминированное поведение при использовании тензоров на устройстве CUDA. Подробнее см. Воспроизводимость.

Примечание

Обратный проход реализован только для src.shape == index.shape.

Предупреждение

Эта функция находится в бета-версии и может измениться в ближайшем будущем.

Параметры

• dim (int) – ось, по которой выполнять индексирование
• index (LongTensor) – индексы элементов для распределения и сворачивания.
• src (Tensor) – исходные элементы для распределения и сворачивания
• reduce (str) – операция сворачивания для применения к не уникальным индексам ("sum", "prod", "mean", "amax", "amin")
• include_self (bool) – включать ли элементы из тензора self в сворачивание

Пример:

>>> src = torch.tensor([1., 2., 3., 4., 5., 6.])
>>> index = torch.tensor([0, 1, 0, 1, 2, 1])
>>> input = torch.tensor([1., 2., 3., 4.])
>>> input.scatter_reduce(0, index, src, reduce="sum")
tensor([5., 14., 8., 4.])
>>> input.scatter_reduce(0, index, src, reduce="sum", include_self=False)
tensor([4., 12., 5., 4.])
>>> input2 = torch.tensor([5., 4., 3., 2.])
>>> input2.scatter_reduce(0, index, src, reduce="amax")
tensor([5., 6., 5., 2.])
>>> input2.scatter_reduce(0, index, src, reduce="amax", include_self=False)
tensor([3., 6., 5., 2.])