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

相关推荐
乐迪信息几秒前
乐迪信息:基于AI算法的煤矿作业人员安全规范智能监测与预警系统
大数据·人工智能·算法·安全·视觉检测·推荐算法
oe10191 小时前
好文与笔记分享 Paris, A Decentralized Trained Open-Weight Diffusion Model
人工智能·笔记·去中心化·多模态
HelloWorld__来都来了1 小时前
Agent S / Agent S2 的架构、亮点与局限
人工智能·架构
JAVA学习通1 小时前
发布自己的 jar 包到 Maven 中央仓库 ( mvnrepository.com )
人工智能·docker·自然语言处理·容器·rocketmq
文火冰糖的硅基工坊1 小时前
[嵌入式系统-107]:语音识别的信号处理流程和软硬件职责
人工智能·语音识别·信号处理
lianyinghhh2 小时前
瓦力机器人-舵机控制(基于树莓派5)
人工智能·python·自然语言处理·硬件工程
dlraba8022 小时前
PyTorch 模型部署实战:用 Flask 搭图像分类 API
pytorch·分类·flask
小殊小殊2 小时前
超越CNN:GCN如何重塑图像处理
图像处理·人工智能·深度学习
康语智能2 小时前
科技赋能成长,小康AI家庭医生守护童真
人工智能·科技
WLJT1231231232 小时前
科技赋能塞上农业:宁夏从黄土地到绿硅谷的蝶变
大数据·人工智能·科技