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

相关推荐
IT_陈寒1 分钟前
Java静态变量初始化坑得我头皮发麻
前端·人工智能·后端
知行产研8 分钟前
白马山领跑、南山矿扩编 31 台,安徽露天矿无人驾驶进入规模化阶段
人工智能
FinelyYang9 分钟前
考试系统简答题 AI 阅卷:覆盖度驱动、异步评分与人机协同落地实践
java·大数据·人工智能
伍树明20 分钟前
mini-swe-agent 本地部署+效果评测记录
人工智能
运维行者_26 分钟前
网络监控工具怎么选?从设备监控到智能运维的能力清单
运维·服务器·开发语言·网络·人工智能·支持向量机·php
清华kenny36 分钟前
安徽亳州中医AI健康辨识仪品牌排名前十之适用场景解析
人工智能·安徽晓得养
郑鱼咚38 分钟前
大胆点,向 AI 要你真正想要的结果
人工智能
信可维40 分钟前
多个 AI 同时改一个仓库:先划边界,再谈并行
人工智能
尺度商业41 分钟前
科技牛错杀与换挡的底层逻辑
人工智能·科技
宅小年1 小时前
Grok Bot 两周实测:把 6 个 Bot 拉进群,我给自己搭了协同小队
人工智能