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

相关推荐
小仙女的小稀罕9 小时前
黑科技重磅更新AI加持高科技AI工具,颠覆场景快准稳极致呈现
人工智能·科技
Rorsion9 小时前
循环神经网络(RNN)
人工智能·rnn·深度学习
jz_ddk9 小时前
[实战] CIC滤波器设计与实现
人工智能·算法·机器学习·数字信号处理·cic滤波器
网管NO.19 小时前
OpenClaw 多模型配置完整教程(WSL2 + Ubuntu)
运维·网络·人工智能·ubuntu
东离与糖宝9 小时前
不用Python!Java+Spring AI 3.x本地RAG系统搭建实战
java·人工智能
找了一圈尾巴9 小时前
OpenClaw 技术架构解析-网关层(下)
人工智能·架构
努力的小白o(^▽^)o9 小时前
简历中关于分类的问题
大数据·人工智能·分类
凌云之程9 小时前
避坑宝典:PyTorch最简安装路径(含CUDA + VSCode + 中文手册)
pytorch·python·conda·安装
Ms_lan9 小时前
体育运动手环训练为何还需要蓝牙网关加持?
人工智能·蓝牙网关·北京桂花网·体育运动监测
skywalk816310 小时前
参考paddlex的图像识别和目标检测,做一个精简的寻物小助手的推理服务器后台
服务器·人工智能·目标检测