PyTorch cat() 函数实现维度拼接

需要保证（除需要合并的维度外）其他维度均相等。参数dim指定需要合并的维度的索引号。

如下例子为合并第三维。

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

输出结果如下：

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

Ref:

Pytorch 维度拼接与维度拆分

总结

以上是生活随笔为你收集整理的PyTorch cat() 函数实现维度拼接的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。