1、使用 None 增加 tensor 的维度
在维度中使用 None 可以在所在维度中增加一维。
2、使用 unsqueeze() 增加 tensor 的维度
torch.unsqueeze(input, dim) → Tensor
torch.Tensor.unsqueeze(dim) → Tensor
- 参数：
- input：input tensor.
- dim：在 dim 所在的索引位置插入一个新的维度，索引范围在 [-input.dim() - 1, input.dim() + 1) 之间。
3、使用 squeeze() 压缩 tensor 的维度
torch.squeeze(input, dim=None) → Tensor
torch.Tensor.squeeze(dim=None) → Tensor
- 参数：
- input：input tensor.
- dim：要压缩的维度索引，如果 dim 为 None，则压缩 input tensor 中所有维度为 1 的维度，如果 dim 指定，则当 dim 索引所在的维度为1时，只压缩该维度，当 dim 索引所在的维度不为1时，不做压缩。