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 people13 分钟前
TensorFlow 基础训练循环(简化版 + 补全代码)
人工智能·python·tensorflow
爱打代码的小林17 分钟前
numpy库数组笔记
笔记·python·numpy
Slaughter信仰18 分钟前
图解大模型_生成式AI原理与实战学习笔记(第四章)
人工智能·笔记·学习
拓端研究室23 分钟前
2025医疗健康行业革新报告:AI赋能、国际化|附170+份报告PDF、数据、可视化模板汇总下载
人工智能·pdf
DisonTangor23 分钟前
iMontage: 统一、多功能、高度动态的多对多图像生成
人工智能·ai作画·开源·aigc
martian66530 分钟前
详解高阶数学领域-信息论与深度学习:互信息在对比学习中的应用
人工智能·深度学习·学习
johnny23335 分钟前
MinerU:理论与实战
人工智能
雪不下37 分钟前
计算机中的数学:概率(6)
人工智能·机器学习·概率论
진영_37 分钟前
深度学习打卡第J1周:ResNet-50算法实战与解析
人工智能·深度学习