本文主要是介绍Pytorch基础（21）-- torch.repeat_interleave()方法，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
分享一下自己目前在维护的Github项目，由于本人博士阶段接触了一个全新的研究方向-----使用机器学习、强化学习、深度学习等方法解决组合优化问题，维护这个项目的目的：
（1）记录自己阅读过的paper，同时分享一下对某些paper的见解；
（2）方便其他的初学者或研究生查阅最新的paper；
⭐️ 项目链接：Welcom to COPZoo
一、前言
嗯……，又是奇奇怪怪的方法，果然，你能想到的功能，大佬们已经实现了。今天要分享的内容是torch.repeat_interleave() 方法。话不多说，Let’s go!
二、方法解析
官方对repeat_interleave方法的介绍及其简单，介绍的越简单，对于使用者来说越难以理解。官方解释：重复一个tensor中元素。嗯……确实简单！
参数
- input：白话就是你要操作的tensor
- repeat：每个元素要重复的次数，可以是int型；也可以是一个Tensor，但要注意的是，这个Tensor进行广播后，要符合给定维度的shape。
- dim：可选参数，你想要沿着哪个维度进行重复。值得注意的是：如果不设置，默认首先将input展开为1维tensor，然后再进行重复。
output_size：可选参数，一般用不到。这里没研究。
三、案例分析
import numpy as np
import torch
3.1 案例1 - repeats为default
t = torch.tensor([[1,2,3],[4,5,6]])
t
t1 = torch.repeat_interleave(input=t, repeats=2)
t1
3.2 案例2 - repeats为int型
repeats=0
t2 = torch.repeat_interleave(input=t, repeats=2, dim=0)
t2
repeats=1
t3 = torch.repeat_interleave(input=t, repeats=2, dim=1)
t3
3.3 案例3 - repeats为tensor型
t4 = torch.repeat_interleave(input=t, repeats=torch.tensor([2,3]), dim=0)
t4
t5 = torch.repeat_interleave(input=t, repeats=torch.tensor([2,3,1]), dim=1)
t5
这篇关于Pytorch基础（21）-- torch.repeat_interleave()方法的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
