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

相关推荐
朱涛的自习室几秒前
预算 1000 美金的开源软件,DeepSeek 只花 1 美金就做完了(Day 6)
android·前端·人工智能
johnsong几秒前
陶哲轩的战争宣言:AI 能否改写数学的未来?
人工智能
桃西西呀4 分钟前
我给网关加了计量埋点,才发现限流配错三个月,下游账单多烧了两倍
人工智能·llm·ai编程
学页官方10 分钟前
把教培训答流程封装成 AI Skill:一次编写,全教研组复用
人工智能·ai
天空属于哈夫克314 分钟前
企业微信 API 二次开发:如何搭建 AI 智能客服系统?
人工智能·企业微信
在所不辞兄16 分钟前
为什么残差连接能够缓解深层神经网络的退化问题,而不仅仅是改善梯度传播?
人工智能·深度学习·神经网络·机器学习
燃烧的枫17 分钟前
Windows 下 YOLO26 的 TensorRT C++ 部署
人工智能·pytorch·windows·计算机视觉·tensorrt·cuda·yolo26
Dawson Zhu20 分钟前
【AI Infra小白入门⑧】PyTorch框架Infra实战指南:Tensor、autograd与性能调试
人工智能·语言模型·架构·aigc·agi
明志数科20 分钟前
遥操作、UMI与头戴视角采集:三类机器人操作数据的工程对比与选型方法
人工智能·机器人
电子元件小说家24 分钟前
2026年秋季电感市场观察:AI算力与汽车电子双轮驱动下的选型与验证指南
人工智能·汽车