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

相关推荐
虾神说D9 小时前
[AI时代码农生存指南]Rust编写CLI 01. CLI的复古轮回
开发语言·人工智能·rust
IT观测9 小时前
轴重检测优选装备 浙江润鑫轴重检测仪稳定可靠
人工智能
秦ぅ时9 小时前
Recraft-V3 技术手册
人工智能·gpt
Cosolar9 小时前
文生图竞技场变局:GPT-Image-2 以 1512 分登顶,多模态格局重塑
人工智能·开源·全栈
博.闻广见9 小时前
AI_线性代数-6.PCA降维详解
人工智能·线性代数
互联网江湖9 小时前
苹果翻开AI眼镜的“生死簿”
人工智能
RInk7oBjo9 小时前
多Agent编排时代 · OpenAI × Anthropic 跨生态协作
人工智能
Yuanxl9039 小时前
PyTorch模型训练全流程详解
人工智能·pytorch·深度学习
码农的神经元9 小时前
从零搭建一个带 GUI 的机器学习建模系统:多模型切换、遗传算法优化与可视化实战复盘
人工智能·机器学习
一楼的猫9 小时前
茄子小说AI辅助智能写作助手:10倍速创作神器
人工智能·学习·机器学习·学习方法·ai写作·迁移学习·集成学习