(PyTorch) repeat方法详解

(PyTorch) repeat方法详解

方法详解

repeat(*sizes): torch.tensor的方法, 沿着指定的维度重复tensor。

传入参数*Sizes为一个 torch.Size 或 int，代表**沿着每一维重复的次数。

例子

>>> x = torch.Tensor([1, 2, 3])
>>> x.repeat(4, 2)1  2  3  1  2  31  2  3  1  2  31  2  3  1  2  31  2  3  1  2  3
[torch.FloatTensor of size 4x6]
>>> x.repeat(4, 2, 1).size()
torch.Size([4, 2, 3])

源张量x为一个$1\times 3$维张量
x.repeat(4, 2)代表第0维重4次，第1维重复2次
返回一个$3\times 6$维张量。

——来自于pytorch中文文档