torch.Tensor.sum_to_size

Tensor.sum_to_size(*size) → Tensor

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

参数

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