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

相关推荐
测试员周周4 分钟前
【AI测试系统】第3篇:AI生成的测试用例太“水”?14年老兵:规则引擎+AI才是王炸组合
人工智能·python·测试
fzil0018 分钟前
自动投递简历 + 面试进度跟踪
人工智能·面试·职场和发展
Raink老师11 分钟前
【AI面试临阵磨枪-34】单 Agent 与多 Agent(Multi-Agent)架构区别、适用场景、挑战
人工智能·ai 面试
LeesonWong12 分钟前
从 PDF 到 MCP:让 AI Agent 按需查询你的简历
人工智能
灵机一物13 分钟前
灵机一物AI原生电商小程序、PC端(已上线)-【AI 技术周报】2026 年 4 月第 4 周|模型、算力、商业化、安全全景梳理
人工智能
redreamSo17 分钟前
一个只有70行的文件,凭什么拿下GitHub 10万星?
人工智能·开源
互联网志17 分钟前
政策赋能校产融合 推动高校科技成果落地生根
大数据·人工智能·物联网
qcx2321 分钟前
Warp源码深度解析(四):AI Agent原生集成——MCP协议、代码索引与Skills系统
人工智能·ai·agent·源码解析·wrap
Narrastory26 分钟前
Note:强化学习(六)
人工智能·深度学习·强化学习
智枢圈36 分钟前
Embedding 与向量数据库
人工智能