torch.reshape用来改变tensor的shape。 torch.reshape(tensor,shape) import torch a=torch.tensor([[[1,2,3],[4,5,6]], [[7,8,9],[10,11,12]]]) print("a的shape:" ...
確定! 回上一頁