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, который для dimension != dim задается его индексом в src, а для dimension = dim — соответствующим значением в index. Если 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.])