讓我以2D矩陣為例:mat = torch.arange(9).view(3,-1)張量([[0,1,2],[3,4,5],[6,7,8 ]])torch.sum(mat,dim = -2)張量([9,12,15])我發現.
確定! 回上一頁