Pytorch中的torch.cat函数

最新推荐文章于 2026-05-02 11:47:18 发布

原创

最新推荐文章于 2026-05-02 11:47:18 发布 · 5.3k 阅读

收录于

当前文章被以下社区和专栏收录：

本文介绍了Pytorch中的torch.cat函数，用于将两个或多个张量沿着特定维度进行拼接。通过示例展示了如何将两个单独的张量以及张量列表进行连接，并提供了简洁的代码实现。

cat是concatnate的意思：拼接，联系在一起。

先说cat( )的普通用法
如果我们有两个tensor是A和B，想把他们拼接在一起，需要如下操作：

C = torch.cat( (A,B),0 )  #按维数0拼接（竖着拼）
C = torch.cat( (A,B),1 )  #按维数1拼接（横着拼）

>>> import torch
>>> A=torch.ones(2,3)    #2x3的张量（矩阵）                                     
>>> A
tensor([[ 1.,  1.,  1.],
        [ 1.,  1.,  1.]])
>>> B=2*torch.ones(4,3)  #4x3的张量（矩阵）                                    
>>> B
tensor([[ 2.,  2.,  2.],
        [ 2.,  2.,  2.],
        [ 2