mindspore.ops.index_select
==========================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/br_base/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/br_base/docs/api/api_python/ops/mindspore.ops.func_index_select.rst
    :alt: 查看源文件


.. py:function:: mindspore.ops.index_select(input, axis, index)

    根据指定轴和索引对输入tensor进行选取，返回一个新tensor。

    .. note::
        - `index` 的值必须在 `[0, input.shape[axis])` 范围内，超出该范围的结果未定义。
        - 返回的tensor和输入tensor的维度数量相同，其第 `axis` 维度的大小和 `index` 的长度相同，其他维度和 `input` 相同。

    参数：
        - **input** (Tensor) - 输入tensor。
        - **axis** (int) - 指定轴。
        - **index** (Tensor) - 指定索引，一维tensor。

    返回：
        Tensor