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

相关文章

Pytorch学习笔记(一):torch.cat()模块的详解
Pytorch学习笔记(二):nn.Conv2d()函数详解
Pytorch学习笔记(三):nn.BatchNorm2d()函数详解
Pytorch学习笔记(四):nn.MaxPool2d()函数详解
Pytorch学习笔记(五):nn.AdaptiveAvgPool2d()函数详解
Pytorch学习笔记(六):view()和nn.Linear()函数详解
Pytorch学习笔记(七):F.softmax()和F.log_softmax函数详解

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: