torch.unsqueeze函数使用示例

y =x = torch.tensor([1, 0])
x = torch.unsqueeze(x, dim=1)
print(x)
y = torch.unsqueeze(y, dim=0)
print(y)


# tensor([[1],
#         [0]])


# tensor([[1, 0]])