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

相关推荐
生信研究猿13 小时前
#P4869.第2题-基于LSTM进行室内温度预测
人工智能·rnn·lstm
IALab-检测行业AI报告生成13 小时前
IACheck 报告AI审核产品更新清单|上周更新(2026.5.4-2026.5.8)
人工智能
Alson_Code13 小时前
Spring Ai Alibaba
java·人工智能·spring
迅利科技13 小时前
CATIA:高端制造的“数字母体”
人工智能·科技·制造
Honey Ro13 小时前
pytorch中的损失函数使用
人工智能·pytorch·深度学习
weixin_4352081613 小时前
大模型 Agent 面试高频100题——基础篇
人工智能·深度学习·自然语言处理·面试·职场和发展·aigc
青稞社区.13 小时前
OpenAI 翁家翌:“启发式学习”的强化学习新范式
人工智能·经验分享·学习·agi
QYR-分析13 小时前
全球及中国固定翼无人机光电吊舱行业发展现状与前景分析
人工智能·无人机
扬帆破浪13 小时前
免费开源AI软件.桌面单机版,可移动的AI知识库,察元 AI桌面版:公司只允许装签名应用 给察元AI打企业内部分发包
人工智能·windows·电脑·知识图谱
深度学习lover13 小时前
<数据集>yolo 桃子识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·桃子识别