torch.Tensor.sum_to_size

Tensor.sum_to_size(*size) → Tensor

Суммирование this тензора по размеру size. size должен быть совместим с размером this тензора.

Параметры

size (int...) – последовательность целых чисел, определяющая форму выходного тензора.