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.unsqueeze(2)，b为（2,3,1）的矩阵
同理，如果a=torch.randn(2,1,3)
则b=a.squeeze(2)=torch.squeeze(a)，b为（2,3）的矩阵。
本文共 293 字，大约阅读时间需要 1 分钟。
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.unsqueeze(2)，b为（2,3,1）的矩阵
同理，如果a=torch.randn(2,1,3)
则b=a.squeeze(2)=torch.squeeze(a)，b为（2,3）的矩阵。
转载于:https://www.cnblogs.com/zhoubiyu/p/10375417.html