concate将矩阵按指定的方向叠加。
cat(arg1,arg2),arg1是需要stack的矩阵,arg2是操作维度。
arg2=0,按行操作,需要除行外的其他维度的长度相同,上下拼
arg2=1,按列操作,左右拼
arg2=3,前后拼。
torch.tensor下的cat()
最新推荐文章于 2024-08-20 13:09:25 发布
concate将矩阵按指定的方向叠加。
cat(arg1,arg2),arg1是需要stack的矩阵,arg2是操作维度。
arg2=0,按行操作,需要除行外的其他维度的长度相同,上下拼
arg2=1,按列操作,左右拼
arg2=3,前后拼。