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

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