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

相关推荐
Python私教7 分钟前
AI回答太冗长?我设计了三段式流式显示让信息层次分明
人工智能
谁似人间西林客11 分钟前
汽车点焊如何走向工艺智能化?AI质量监控已成为主流解决方案
人工智能·汽车
2601_9567436817 分钟前
上海大模型应用开发技术路径全解析:从架构选型到落地约束
人工智能·软件工程
云天AI实战派17 分钟前
AI智能体总是跑偏怎么办?ChatGPT/API 调用排查指南:从工具路由到语音闭环的全流程修复手册
人工智能·chatgpt·aigc
逐米时代27 分钟前
四川制造企业智改数转怎么申报?本地化AI项目落地一般分5步
大数据·人工智能
weixin_5536544829 分钟前
有没有一种可能,现在的大语言模型已经发展得接近极限了?
人工智能·语言模型·大模型
GISer_Jing29 分钟前
基于 GitHub Actions 端到端工程化落地——AI全栈项目实战案例
人工智能·github
图灵农场29 分钟前
SpringAI实用-RAG
人工智能
LienJack32 分钟前
AI 架构设计有点菜,我写了个 Skill 给它补课
人工智能·架构