pytorch中repeat和repeat_interleave

豪言成笑谈

已于 2022-05-20 15:18:49 修改

阅读量3.4k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： pytorch基础文章标签： pytorch python 人工智能

于 2021-12-09 11:13:55 首次发布

本文链接：https://blog.csdn.net/weixin_43559672/article/details/121809173

pytorch基础专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了PyTorch中`repeat`和`repeat_interleave`两个函数的区别。`repeat`函数会将张量在指定维度上复制并拼接，而`repeat_interleave`则会在某一维度上将元素重复，使得相邻元素相同。通过示例代码展示了两者的不同效果，帮助理解这两个操作在实际应用中的差异。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

repeat的参数是每一个维度上重复的次数，repeat_interleave的参数是重复的次数和维度。
repeat相当于将该张量复制，然后在某一维度concat起来，而repeat_interleave是将张量中的元素沿某一维度复制n次，即复制后的张量沿该维度相邻的n个元素是相同的。
在这里插入图片描述
例子：

a = np.arange(0, 10)
a = a.reshape(2, -1)
a = torch.from_numpy(a)
b = a.repeat(2, 1)
print(b)
c = a.repeat_interleave(2, 1)
print(c)
d = a.repeat(1, 2)
print(d)
e = a.repeat_interleave(2, 0)
print(e)