torch 数组

 

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