PyTorch中torch.cat

版权声明：本文为博主原创文章，欢迎大家转载，但是要注明我的文章地址。 https://blog.csdn.net/program_developer/article/details/82112601

import torch
a = torch.ones([1,2])
print(a)
# tensor([[1., 1.]])

b = torch.ones([1,2])
print(b)
# tensor([[1., 1.]])

print(torch.cat([a,b],1))
# tensor([[1., 1., 1., 1.]])

torch.cat就是将a,b 按行放在一起，如果第二个参数是0，则按列放在一起。