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

相关推荐
NAGNIP8 小时前
轻松搞懂全连接神经网络结构!
人工智能·算法·面试
moshuying9 小时前
别让AI焦虑,偷走你本该有的底气
前端·人工智能
董董灿是个攻城狮10 小时前
零基础带你用 AI 搞定命令行
人工智能
喝拿铁写前端12 小时前
Dify 构建 FE 工作流:前端团队可复用 AI 工作流实战
前端·人工智能
阿里云大数据AI技术13 小时前
阿里云 EMR Serverless Spark + DataWorks 技术实践:引领企业 Data+AI 一体化转型
人工智能
billhan201613 小时前
MCP 深入理解:协议原理与自定义开发
人工智能
Jahzo13 小时前
openclaw桌面端体验--ClawX
人工智能·github
billhan201613 小时前
Agent 开发全流程:从概念到生产
人工智能
threerocks13 小时前
过了个年,AI 圈变天了?但没人告诉你为什么
人工智能