从2*1演化而来，
2行1列就是2*1
2*2*1
就是最后维度只有1个数字，这个数字
import torch
bb=torch.Tensor([[[1],[2]],[[2],[3]]])
print(bb.shape)
exit()bb=bb.squeeze(2)
降个维度，就是2*2了。
torch 数组
猜你喜欢
转载自blog.csdn.net/jacke121/article/details/104726312
今日推荐
周排行