torch.cat可以直接用來連接多個tensor，不需要兩個兩個之間相互concatenate。話不多說，下面寫個例程演示一下：

In [1]: import torchIn [2]: a = torch.ones(3,4)In [3]: b = 2*torch.ones(3,4)In [4]: c = 3*torch.ones(3,4)In [5]: d = torch.cat((a,b,c),axis = 1)In [6]: e = torch.cat((a,b,c),axis = 0)In [7]: d Out[7]: tensor([[1., 1., 1., 1., 2., 2., 2., 2., 3., 3., 3., 3.],[1., 1., 1., 1., 2., 2., 2., 2., 3., 3., 3., 3.],[1., 1., 1., 1., 2., 2., 2., 2., 3., 3., 3., 3.]])In [8]: e Out[8]: tensor([[1., 1., 1., 1.],[1., 1., 1., 1.],[1., 1., 1., 1.],[2., 2., 2., 2.],[2., 2., 2., 2.],[2., 2., 2., 2.],[3., 3., 3., 3.],[3., 3., 3., 3.],[3., 3., 3., 3.]])

總結

以上是生活随笔為你收集整理的torch.cat同时连接多个tensor的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。