numpy的repeat和pytorch的repeat区别

在项目中发现这两个同名的函数的效果不一样,下面简单介绍:

numpy的repeat

cpp 复制代码
a = np.array([1,2,3])  
print('a', a)  ## 结果为123
b = a.repeat(3)
print('b')  ## 结果为111222333

pytorch的repeat和repeat_interleave

cpp 复制代码
labels = np.array([1,2,3])
labels = torch.tensor(labels, dtype=torch.float32)
labels1 = labels.repeat(3)  
print(labels1)  ##结果为123123123
labels2 = labels.repeat_interleave(3, dim=0)
print(labels2 )  ##结果为111222333

numpy没有repeat_interleave函数。

相关推荐
智算菩萨12 小时前
Pip与第三方库:一行命令安装 AI 能力
人工智能·pip
serve the people12 小时前
TensorFlow 基础训练循环(简化版 + 补全代码)
人工智能·python·tensorflow
爱打代码的小林12 小时前
numpy库数组笔记
笔记·python·numpy
Slaughter信仰13 小时前
图解大模型_生成式AI原理与实战学习笔记(第四章)
人工智能·笔记·学习
拓端研究室13 小时前
2025医疗健康行业革新报告:AI赋能、国际化|附170+份报告PDF、数据、可视化模板汇总下载
人工智能·pdf
DisonTangor13 小时前
iMontage: 统一、多功能、高度动态的多对多图像生成
人工智能·ai作画·开源·aigc
martian66513 小时前
详解高阶数学领域-信息论与深度学习:互信息在对比学习中的应用
人工智能·深度学习·学习
johnny23313 小时前
MinerU:理论与实战
人工智能
雪不下13 小时前
计算机中的数学:概率(6)
人工智能·机器学习·概率论
진영_13 小时前
深度学习打卡第J1周:ResNet-50算法实战与解析
人工智能·深度学习