(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 3
1 2 3 1 2 3
1 2 3 1 2 3
1 2 3 1 2 3
[torch.FloatTensor of size 4x6]
>>> x.repeat(4, 2, 1).size()
torch.Size([4, 2, 3])
源张量x为一个
1
×
3
1\times 3
1×3维张量
x.repeat(4, 2)代表第0维重4次,第1维重复2次
返回一个
3
×
6
3\times 6
3×6维张量。
——来自于pytorch中文文档