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右。
参考这篇文章,讲的比较好
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右。