【基于PyTorch】torch.unsqueeze() 使用

• 输入：

import torch

x = torch.tensor([1, 2, 3, 4])
print(torch.unsqueeze(x, 0))
print("**************")
print(torch.unsqueeze(x, 1))

• 输出：

tensor([[1, 2, 3, 4]])
**************
tensor([[1],
        [2],
        [3],
        [4]])