index_select

该OP沿着指定轴 axis 对输入 x 进行索引，取 中指定的相应项，创建并返回到一个新的Tensor。这里 index 是一个 1-D Tensor。除 轴外，返回的Tensor其余维度大小和输入 x 相等 ， axis 维度的大小等于 的大小。

参数：

返回：

-Tensor: 返回一个数据类型同输入的Tensor。