PyTorch cat() 函數(shù)實(shí)現(xiàn)維度拼接

需要保證（除需要合并的維度外）其他維度均相等。參數(shù)dim指定需要合并的維度的索引號(hào)。

如下例子為合并第三維。

import torch a = torch.rand(2, 10, 30) b = torch.rand(2, 10, 40)c = torch.cat([a, b], dim=2) print(c.shape)

輸出結(jié)果如下：

>>> c.shape torch.Size([2, 10, 70])

Ref:

Pytorch 維度拼接與維度拆分

總結(jié)

以上是生活随笔為你收集整理的PyTorch cat() 函数实现维度拼接的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。