pytorch 的 cat函数的详解

最新推荐文章于 2024-04-18 19:45:25 发布

Vertira

最新推荐文章于 2024-04-18 19:45:25 发布

阅读量855

点赞数

分类专栏： pytorch 文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/Vertira/article/details/131484743

版权

pytorch 专栏收录该内容

86 篇文章 20 订阅

订阅专栏

cat是concatnate的意思，也就是说进行张量的拼接

实例：

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

>>> A2=torch.cat((A,A1),0)  #按维数0（行）拼接
>>> A2
tensor([[1., 1., 1.],
        [1., 1., 1.],
        [2., 2., 2.],
        [2., 2., 2.],
        [2., 2., 2.],
        [2., 2., 2.]])
>>>

>>> A.shape
torch.Size([2, 3])
>>> A1.shape
torch.Size([4, 3])
>>> A2.shape
torch.Size([6, 3])
>>>