方法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)

與50位技術專家面對面20年技術見證，附贈技術全景圖

總結

以上是生活随笔為你收集整理的PyTorch基础（11）----- torch.sum()方法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。