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