- 输入:
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]])
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]])