pytorch | tensor維度中使用 None

• 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())

• 結果

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

• 從實驗結果可以看出， None 功能類似torch.unsqueeze(),方便擴展維度，而不改變數據排列順序。