torch.repeat()

torch.repeat函数

函数功能

torch.tensor.repeat()函数可以对张量进行重复扩充
1) 当参数只有两个时：（行的重复倍数，列的重复倍数），1表示不重复。
2) 当参数有三个时：（通道数的重复倍数，行的重复倍数，列的重复倍数），1表示不重复。、

 example：

>>> x = torch.tensor([1, 2, 3])
>>> x.repeat(4, 2)
tensor([[ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3]])
>>> x.repeat(4, 2, 1).size()
torch.Size([4, 2, 3])

更多更细，看博客：torch.repeat函数