/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/mint/mindspore.mint.func_index_select.rst
25 строк
1 KB
wang_ziqi
Add mint.index_select
25 май 2024, 12:32
25 май 2024, 12:32
7412ca5
Код
Авторство
О чём код?
mindspore.mint.index_select ============================ .. py:function:: mindspore.mint.index_select(input, dim, index) 返回一个新的Tensor，该Tensor沿维度 `dim` 按 `index` 中给定的索引对 `input` 进行选择。 返回的Tensor和输入Tensor( `input` )的维度数量相同，其第 `dim` 维度的大小和 `index` 的长度相同；其它维度和 `input` 相同。 .. note:: index的值必须在 `[0, input.shape[dim])` 范围内，超出该范围结果未定义。 参数： - **input** (Tensor) - 输入Tensor。 - **dim** (int) - 根据索引进行选择的维度。 - **index** (Tensor) - 包含索引的一维Tensor。 返回： Tensor，数据类型与输入 `input` 相同。 异常： - **TypeError** - `input` 或 `index` 的类型不是Tensor。 - **TypeError** - `dim` 的类型不是int。 - **ValueError** - `dim` 值超出范围[-input.ndim, input.ndim - 1]。 - **ValueError** - `index` 不是一维Tensor。