torch.Tensor.sum_to_size

Tensor.sum_to_size(*size) → Tensor

Sum this tensor to size. size must be broadcastable to this tensor size.

Parameters

size (int...) – a sequence of integers defining the shape of the output tensor.