摘要： import torch torch中的squeeze与unsqueeze作用是去除/添加维度为1的行 例如，a=torch.randn(2,3) 那么b=a.unsqueeze(0)，b为（1,2,3）的矩阵 类似的，b=a.unsqueeze(1)，b为（2,1,3）的矩阵 b=a.unsque 阅读全文