torch.Tensor.sum_to_size

Tensor.sum_to_size(*size) → Tensor

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

Параметры:

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