torch.repeat 和 torch.repeat_interleave

** 结论
torch.repeat: 输入张量的从后往前的后面维度对应按照repeat中大小进行repeat操作（所以 输入张量维度>= repeat维度）。 假设输入张量为（a,b,c）,repeat(x,y),则为b维度repeat x倍，c维度repeat y倍；最终输出维度为（a, bx, cy）
torch.repeat_interleave: 输入张量按照指定维度进行扩展，假设输入张量为（2,2），torch.repeat_interleave(y, 3, dim=1)， 原输入张量大小为（2,2），则在维度1扩展3倍，最终为（2，6）。如果没有指定dim，则会将输入拉张开为1维向量再进行扩展

1.torch.repeat

x = torch.tensor([1, 2, 3])
x.repeat(4, 2), x.repeat(4, 2).shape, x.repeat(4, 2, 1).shape, x.repeat(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]]),

torch.Size([4, 6]),
torch.Size([4, 2, 3]),
tensor([1, 2, 3, 1, 2, 3]))

2.torch.repeat_interleave

x = torch.tensor([1, 2, 3])
x.repeat_interleave(2)
tensor([1, 1, 2, 2, 3, 3])
y = torch.tensor([[1, 2], [3, 4]])
torch.repeat_interleave(y, 2)
tensor([1, 1, 2, 2, 3, 3, 4, 4])
torch.repeat_interleave(y, 3, dim=1)
tensor([[1, 1, 1, 2, 2, 2],
[3, 3, 3, 4, 4, 4]])
torch.repeat_interleave(y, torch.tensor([1, 2]), dim=0)
tensor([[1, 2],
[3, 4],
[3, 4]])

__EOF__

本文作者：userName
本文链接：https://www.cnblogs.com/pyclq/p/15069562.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！