来源：https://blog.csdn.net/qq_50001789/article/details/120315215
torch.index_select(input, dim, index, *, out=None) → Tensor
功能
选择根据给定的index和dim在input中选择张量数据，相当于更高级的索引功能。
输入
input：需要索引的张量数组
dim：索引维度(沿dim维度进行索引)
index：索引值，可以是单个数字、也可以是一个序列(一维序列)
注意
- 返回的张量数组与原始的张量数组具有相同的维数，这里与直接进行索引有区别，具体见案例代码；
dim维度的尺寸大小与index的长度相同，其他尺寸大小与原始张量中的尺寸相同index：维数必须小于等于1