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

相关推荐
大知闲闲哟11 分钟前
深度学习J7周 对于ResNeXt-50算法的思考
人工智能·深度学习
Kai HVZ13 分钟前
《机器学习》——逻辑回归(过采样)
人工智能·机器学习·逻辑回归
雾江流18 分钟前
MagicMirror 1.0.0 | 基于AI的面部替换、发型和服装搭配应用,无需GPU支持
人工智能·软件工程
qq_2739002324 分钟前
pytorch张量高级索引介绍
人工智能·pytorch·python
RacheV+TNY26427833 分钟前
电商数据API接口的智能化与自动化发展探索
网络·人工智能·python·自动化·api
听吉米讲故事40 分钟前
AI新闻自动化:使用Tavily Search API构建AI新闻总结助手
人工智能·搜索引擎·自动化
程序员陆通1 小时前
2024年大型语言模型(LLMs)的发展回顾
人工智能·语言模型·自然语言处理
Shiyuan71 小时前
【IEEE冠名会议】2025年IEEE第二届深度学习与计算机视觉国际会议(DLCV 2025)
人工智能·深度学习·计算机视觉
是十一月末1 小时前
Opencv实现Sobel算子、Scharr算子、Laplacian算子、Canny检测图像边缘
人工智能·python·opencv·计算机视觉
MYT_flyflyfly1 小时前
计算机视觉之三维重建-摄像机标定
人工智能·算法·计算机视觉