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函数。

相关推荐
写点啥呢几秒前
车机 Android 开机优化复盘:我怎么和 AI 一起把问题定位到 SystemUI
android·人工智能
AI客栈2 分钟前
云原生 AI 平台安全设计
人工智能
苏州邦恩精密4 分钟前
GOM三维扫描在制造中的真实价值:让“修模”从经验动作变成数据动作
人工智能·科技·机器学习·3d·自动化·制造
邵宇然8 分钟前
Rust 系统编程实战:从所有权模型到零成本抽象的工程落地
人工智能
大山佬15 分钟前
传感器驱动开发:从硬件时序到 Linux IIO 子系统
人工智能
mit6.82416 分钟前
计算机小白自学的两年
人工智能
龙腾AI白云19 分钟前
数字孪生和世界模型,二者的技术边界正在慢慢融合吗?
人工智能·django·知识图谱
蓦然回首却已人去楼空20 分钟前
【转载+大量补充】深入理解深度学习中常见激活函数
人工智能·深度学习
Swift社区20 分钟前
当 AI 接管游戏世界:鸿蒙游戏 Workspace Runtime 架构揭秘
人工智能·游戏·harmonyos
小t说说23 分钟前
技术观察:从职坐标看一家IT培训机构的课程体系与AI教学工具
大数据·人工智能