squeeze()官方文档
简而言之,squeeze()用于压缩维度,unsqueeze()用于扩充维度
Pytorch中将图像输入CNN进行预测之前,进行unsqueeze操作:
img=torch.unsqueeze(img, dim=0)
是因为网络的接收输入是一个mini-batch,image unsqueeze后第一个维度是留给batch size的
squeeze()官方文档
简而言之,squeeze()用于压缩维度,unsqueeze()用于扩充维度
Pytorch中将图像输入CNN进行预测之前,进行unsqueeze操作:
img=torch.unsqueeze(img, dim=0)
是因为网络的接收输入是一个mini-batch,image unsqueeze后第一个维度是留给batch size的