letmetakea2Dmatrixasexample: mat=torch.arange(9).view(3,-1) tensor([[0,1,2], [3,4,5], [6,7,8]]) torch.sum(mat,dim=-2) tensor([9,12,15]) ...
確定! 回上一頁