其他分享
首页 > 其他分享> > PyTorch的torch.cat用法

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右。

标签:torch,cat,PyTorch,拼接,tencent,按维数
来源: https://www.cnblogs.com/lanhongfu/p/16508325.html