torch.Tensor.sum_to_size

Tensor.sum_to_size(*size) → Tensor

将 this 张量求和到 size。 size 必须可以广播到 this 张量大小。

参数

size (int...) – 定义输出张量形状的整数序列。