squeeze() 和 unsqueeeze()

tensor.squeeze(index) 字面意思:就是在index处,如果维度为1,就删掉这个维度。如果此处维度不为1,则不操作

tensor.unsqueeeze(index) :就是在index处,增加一个维度为1的维度;

注意 比如tensor a 的维度是(2,1,3)

a.squeeze(1)   的维度是(2,3),但a的维度仍然是(2,1,3)。就是squeeze和unsqueeze不会对tensor本身产生修改,修改后要赋值给别的变量才能够保存。

如 b = a.squeeze(1)

你可能感兴趣的:(深度学习模型,模型模块学习,深度学习,python)