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

相关推荐
shujudang几秒前
智能数据分析平台有哪些?五个主流平台的数据模型与选型对比
大数据·人工智能·数据挖掘·数据分析
snhna1 分钟前
GPT Researcher 源码拆解:研究型 Agent 如何检索、筛选证据与生成报告
人工智能
看-是灰机8 分钟前
Milvus向量数据库
linux·运维·人工智能·机器学习·milvus
新新学长搞科研8 分钟前
【厦门大学主办丨IEEE出版】第六届人工智能、机器人和通信国际会议(ICAIRC 2026)
人工智能·机器人·信息与通信
LiuPig刘皮哥11 分钟前
xiaoke.ai 小可爱技术路线图解:3D 陪玩、虚拟陪伴、透明桌宠
人工智能·3d·开源
weixin_5498083612 分钟前
AI HR进入日常工作:AI原生如何改变业务交接,eRoad iBuilder如何参与
大数据·人工智能·ai-native
坐吃山猪13 分钟前
AI复刻网站功能初体验
人工智能·开发
初学者↑15 分钟前
知识图谱+智能体的智慧旅游问答系统
人工智能·知识图谱·旅游
科技每日热闻17 分钟前
企业使用大模型顾虑数据泄露,推荐选用哪些安全可靠的生成式 AI 平台?
人工智能·ai
SelectDB技术团队21 分钟前
VARIANT 列接入网关 JSON 日志:建表、导入、子列查询与物化的完整步骤
大数据·数据库·人工智能·elasticsearch·搜索引擎·json·apache doris