cat是concatnate的意思,也就是说进行 张量 的拼接
实例:
>>> import torch
>>> A=torch.ones(2,3) #2x3的张量(矩阵)
>>> A
tensor([[1., 1., 1.],
[1., 1., 1.]])
>>> A1=2*torch.ones(4,3) #4x3的张量(矩阵)
>>> A1
tensor([[2., 2., 2.],
[2., 2., 2.],
[2., 2., 2.],
[2., 2., 2.]])
>>> A2=torch.cat((A,A1),0) #按维数0(行)拼接
>>> A2
tensor([[1., 1., 1.],
[1., 1., 1.],
[2., 2., 2.],
[2., 2., 2.],
[2., 2., 2.],
[2., 2., 2.]])
>>>
>>> A.shape
torch.Size([2, 3])
>>> A1.shape
torch.Size([4, 3])
>>> A2.shape
torch.Size([6, 3])
>>>