1、一种方便的扩维方法:加None。示例如下:
>>>torch.arange(start=0, end=10)
>>>tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>>A[:, None]
>>>tensor([[0],
[1],
[2],
[3],
[4],
[5],
[6],
[7],
[8],
[9]])
>>>A[:, None].size()
>>>torch.Size([10, 1])
>>>A[None].size()
>>>torch.Size([1, 10])
2.创建一个空tensor
A=torch.tensor([]).cuda()
空tensor可以和已经存在的tensor cat
A=torch.cat([A, output], 0)