Pytorch学习笔记(一)：torch.cat()模块的详解

1.What is torch.cat()

cat是concatnate的意思：拼接，联系在一起。所以显而易见就是拼接两个tensor的意思

2.How to use torch.cat()

C = torch.cat( (A,B),0 ) #按维数0拼接（竖着拼）

C = torch.cat( (A,B),1 ) #按维数1拼接（横着拼）

3.Actual code of torch.cat()

import torch
A=torch.zeros(3,4)
B=torch.ones(3,4)
print(A)
print(B)
C = torch.cat((A,B),0)  #按维数0拼接（竖着拼,需要列数相同)
D = torch.cat((A,B),1)  #按维数1拼接（横着拼,需要行数相同）
print(C)
print(D)

THE RESULT IS: