[已解决]torch里面sum函数不太明白

季夏。。。

A = torch.arange(20*2).reshape(2,5,4)
A_sum_axis1 = A.sum(axis=1)

为什么这个A_sum_axis1是tensor([[ 40,  45,  50,  55],[140, 145, 150, 155]])？
A是tensor
     ([[[ 0,  1,  2,  3],
         [ 4,  5,  6,  7],
         [ 8,  9, 10, 11],
         [12, 13, 14, 15],
         [16, 17, 18, 19]],

        [[20, 21, 22, 23],
         [24, 25, 26, 27],
         [28, 29, 30, 31],
         [32, 33, 34, 35],
         [36, 37, 38, 39]]])

最佳答案

月排行榜 / 总排行榜

dolly_yos2

2022-10-8 21:49:56

看看这个有没有帮助
https://pytorch.org/docs/stable/generated/torch.sum.html#torch.sum

跳转到最佳答案楼层

dolly_yos2

看看这个有没有帮助
https://pytorch.org/docs/stable/generated/torch.sum.html#torch.sum