PyTorch的torch.cat用法

参考这篇文章,讲的比较好
https://cloud.tencent.com/developer/article/1725102

C=torch.cat((A,B),0)就表示按维数0(行)拼接A和B,也就是竖着拼接,A上B下。
C=torch.cat((A,B),1)就表示按维数1(列)拼接A和B,也就是横着拼接,A左B右。

posted @ 2022-07-22 23:59  猪猪猪猪侠  阅读(61)  评论(0编辑  收藏  举报