import torch a = torch.tensor([[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]]) print(a.size()) c = torch.sum(a,dim=1) print(c,c.size()) b ...
確定! 回上一頁