PyTorch基础（11）----- torch.sum()方法

方法1详解

torch.sum(input, *, dtype=None) → Tensor

• input：输入的张量

案例

x = torch.randn(1, 3)
print(x)
y = torch.sum(x)
print(y)

方法2详解

torch.sum(input, dim, keepdim=False, *, dtype=None) → Tensor

• input：输入的张量
• dim：求和的维度，如果dim=1，则按行求和；如果dim=0，则按列求和
• keepdim：默认为False，若keepdim=True，则返回的Tensor除dim之外的维度与input相同

案例

• 案例1

x = torch.arange(0, 12).view(3, 4)
print(x)
y = torch.sum(x, dim=1)
print(y)

• 案例2

x = torch.arange(0, 12).view(3, 4)
print(x)
y = torch.sum(x, dim=0)
print(y)

• 案例3

x = torch.arange(0, 12).view(3, 4)
print(x)
y = torch.sum(x, dim=0, keepdim=True)
print(y)