PyTorch中使用 Noneunsqueeze()squeeze() 改变 tensor 的维度

Posted 2021-12-14 魏晓蕾

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时，不做压缩。