因为在dim上加和之后,dim=1,所以可以直接删去。import torcha = torch.tensor([[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]])print(a.size())c = torch.sum(a,dim=1)...
確定! 回上一頁