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

相关推荐
上海云盾-小余1 小时前
警惕 “伪装型” CC 攻击!通过日志分析识别异常请求,让恶意访问无所遁形
人工智能·安全·架构
Coding茶水间1 小时前
基于深度学习的面部口罩检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
WenGyyyL1 小时前
深度学习数学基础(一)——线性代数、线性代数和微积分
人工智能·深度学习·线性代数
musk12121 小时前
YOLOv8n模型微调全指南:从环境搭建到技能储备 (内容由 AI 生成)
人工智能·yolo
JeffyW1 小时前
Claude Agent 长时间运行实践指南
人工智能
爱笑的眼睛111 小时前
Flask应用API深度开发:从单体架构到微服务设计模式
java·人工智能·python·ai
LHZSMASH!1 小时前
基于动态图卷积与时间自注意力的EEG情绪识别混合网络——深度技术解析
人工智能·深度学习
彼岸花开了吗1 小时前
构建AI智能体:四十、K-Means++与RAG的融合创新:智能聚类与检索增强生成的深度应用
人工智能·python
趁你还年轻_1 小时前
检索增强生成 RAG
人工智能
蒙奇·D·路飞-1 小时前
大模型领域的较量与发展趋势:从技术突破到产业变革
人工智能