从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了。
从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了。