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 分钟前
智能体工作流:从架构设计到行业落地
人工智能
老吴学AI7 分钟前
系列报告一、《The State of Enterprise AI 2025》By OpenAI 中文简介
人工智能
m0_6924571022 分钟前
缺陷检测-药品成品率检测
人工智能·缺陷检测
下海fallsea38 分钟前
韩国零食正在占领俄罗斯
大数据·人工智能
dulu~dulu1 小时前
机器学习试题总结
人工智能·决策树·机器学习·支持向量机·学习笔记·线性回归·集成学习
Coder_Boy_1 小时前
基于SpringAI的企业级智能教学考试:基础资源构建模块
人工智能·spring boot
北邮刘老师1 小时前
【智能体互联协议解析】AIP/ACPs如何实现“自主互联,协商互通,独立自治”
网络·人工智能·大模型·智能体·智能体互联网
Coder个人博客1 小时前
Transformers分词器模块深度分析
人工智能·自动驾驶·transformer
简简单单OnlineZuozuo1 小时前
视觉模型的偏见和捷径测试:来自真实世界实验的实用测试套件
人工智能·深度学习·安全·图像识别·banana
Coder_Boy_1 小时前
基于SpringAI的企业级智能教学考试平台模块二:核心执行模块内容
人工智能·spring boot