AndyChingccc的博客

各种情况下pytorch 如何保存/加载模型

官方文档：torch.squeeze — PyTorch 1.10 documentationtorch.squeeze(input, dim=None, *, out=None) → Tensorsqueeze()函数对输入张量进行维度压缩后返回一个tensor（张量），其中 input 中大小为1的所有维都已删除。1. 例子：如果 input 的形状为 (A×1×B×C×1×D)，那么返回的tensor的形状则为 (A×B×C×D)当给定 dim 时，那么只在给定的维度（dimensi