pytorch | tensor维度中使用 None_jmucvm的博客

• None可以在所处维度中多一维，具体实现。

import torch
tensor = torch.randn(3, 4)
print('tensor size:', tensor.size())
tensor_1 = tensor[:, None]
print('tensor_1 size:', tensor_1.size())
tensor_2 = tensor[:, :, None]
print('tensor_2 size', tensor_2.size())
1
2
3
4
5
6
7

• 结果

tensor size: torch.Size([3, 4])
tensor_1 size: torch.Size([3, 1, 4])
tensor_2 size torch.Size([3, 4, 1])
1
2
3

• 从实验结果可以看出， None 功能类似torch.unsqueeze(),方便扩展维度，而不改变数据排列顺序。