季夏。。。 发表于 2022-10-8 17:01:33

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],])？
A是tensor
   ([[[ 0,1,2,3],
         [ 4,5,6,7],
         [ 8,9, 10, 11],
         ,
         ],

      [,
         ,
         ,
         ,
         ]])

dolly_yos2 发表于 2022-10-8 21:49:56

看看这个有没有帮助
https://pytorch.org/docs/stable/generated/torch.sum.html#torch.sum
页: [1]
查看完整版本: torch里面sum函数不太明白