Pytorch中的torch.cat()函数

今天在看代码的过程中，遇到了

torch.cat((data_hazy, clean_image),0)

当时猜测，应该是拼接函数，果不其然。cat是concatenate的缩写，而concatenate指连接，拼接的意思。

如果，A和B是两个tensor，那么

torch.cat((A, B), 0) #竖着拼接
torch.cat((A, B), 1) #横着拼接

上面成立的前提条件是：竖着拼接（列数一致），横着拼接（行数一致），如下所示：

import torch
A = torch.ones(2,3)
B = torch.ones(4,3)
C = torch.cat((A, B), 0)
print(C)

同样的，按行拼接，如下所示：

import torch
A = torch.ones(2,3)
B = torch.ones(2,6)
C = torch.cat((A, B), 1)
print(C)