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

相关推荐
weixin_468466852 分钟前
YOLOv11结构解析及源码复现
人工智能·深度学习·yolo·目标检测·计算机视觉·图像识别·yolov11
攻城狮7号7 分钟前
京东AI数字人直播服务对商家免费开放:一场“免费午餐”的背后
人工智能·电商·ai数字人·京东直播服务
炫饭第一名16 分钟前
前端玩转 AI 应用开发|SSE 协议与JS中的流式处理🌊
前端·人工智能·程序员
leo_23222 分钟前
AI私教--SMP(软件制作平台)开发的第一个产品(小视频)
人工智能·smp(软件制作平台)·软件开发工具·应用系统
T___T22 分钟前
通过 MCP 让 AI 读懂你的 Figma 设计稿
前端·人工智能
悟纤28 分钟前
Suno 创作Rap音乐李白的《将进酒》,音乐风格音乐无所不能 | 从零开始用Suno Ai | 第5篇
人工智能·suno ai·suno api·ai music
小小工匠32 分钟前
LLM - Google 5-Day AI Agents Intensive
人工智能·ai agents
雨大王51234 分钟前
AI视觉检测怎么选?技术原理、行业应用与解决方案解析
人工智能·计算机视觉·视觉检测
ziwu35 分钟前
【垃圾识别系统】Python+TensorFlow+Vue3+Django+人工智能+深度学习+卷积网络+resnet50算法
人工智能·深度学习·图像识别
Elastic 中国社区官方博客1 小时前
Elasticsearch:数据脱节如何破坏现代调查
大数据·数据库·人工智能·elasticsearch·搜索引擎·ai·全文检索