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時，不做壓縮。