Pytorch中的repeat()函数

Jackie_Wang11

已于 2022-06-13 15:17:35 修改

阅读量2k

点赞数 4

分类专栏： pytorch CV python 文章标签： pytorch 深度学习 python

于 2022-05-01 09:42:36 首次发布

本文链接：https://blog.csdn.net/jq_98/article/details/124523041

版权

python 同时被 3 个专栏收录

60 篇文章 3 订阅

订阅专栏

20 篇文章 0 订阅

订阅专栏

pytorch

5 篇文章 0 订阅

订阅专栏

pytorch中的repeat()函数可以对张量进行复制。

当参数只有一个时，参数表示在列方向上复制的次数
当参数只有两个时，第一个参数表示在行方向上复制的次数，第二个参数表示在列方向上复制的次数。
当参数有三个时，第一个参数表示在通道数方向上复制的次数，第二个参数表示在行方向上复制的次数，第三个参数表示在列方向上复制的次数。

接下来我们举一个例子来直观理解一下：

>>> x = torch.tensor([6,7,8])
>>> x.repeat(4)
tensor([[6, 7, 8, 6, 7, 8, 6, 7, 8, 6, 7, 8]])


>>> x = torch.tensor([6,7,8])
>>> x.repeat(4,2)
tensor([[6, 7, 8, 6, 7, 8],
        [6, 7, 8, 6, 7, 8],
        [6, 7, 8, 6, 7, 8],
        [6, 7, 8, 6, 7, 8]])
>>> x = torch.tensor([6,7,8])
>>> x.repeat(4,1)
tensor([[6, 7, 8],
        [6, 7, 8],
        [6, 7, 8],
        [6, 7, 8]]) 
      
>>> x.repeat(4,2,1)
tensor([[[6, 7, 8],
         [6, 7, 8]],

        [[6, 7, 8],
         [6, 7, 8]],

        [[6, 7, 8],
         [6, 7, 8]],

        [[6, 7, 8],
         [6, 7, 8]]])
>>> x.repeat(4,2,1).size()
torch.Size([4, 2, 3])