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

相关推荐
DJ.马7 小时前
如何在环境里同时配置tensorflow和pytorch共存
人工智能·pytorch·tensorflow
星期五不见面7 小时前
机器人学习!(二)ROS-基于Gazebo项目-YOLO(3)2026/01/13
人工智能·学习·机器人
bst@微胖子7 小时前
HuggingFace项目实战之使用Trainer执行训练
人工智能·机器学习
d0ublεU0x007 小时前
注意力机制与transformer
人工智能·深度学习·transformer
凤希AI伴侣7 小时前
凤希AI提出:FXPA2P - 当P2P技术遇上AI,重新定义数据与服务的边界
人工智能·凤希ai伴侣
腾迹7 小时前
2026年企业微信SCRM系统服务推荐:微盛·企微管家的AI私域增长方案
大数据·人工智能
寰宇视讯7 小时前
脑科技走进日常 消费级应用开启新蓝海,安全与普惠成关键
人工智能·科技·安全
云卓SKYDROID7 小时前
无人机电机模块选型与技术要点
人工智能·无人机·遥控器·高科技·云卓科技
小酒星小杜7 小时前
在AI时代,技术人应该每天都要花两小时来构建一个自身的构建系统 - 总结篇
前端·vue.js·人工智能
云卓SKYDROID7 小时前
无人机螺旋桨材料与技术解析
人工智能·无人机·高科技·云卓科技·技术解析、