27 lines
1.4 KiB
ReStructuredText
27 lines
1.4 KiB
ReStructuredText
mindspore.numpy.take
|
||
=================================
|
||
|
||
.. py:function:: mindspore.numpy.take(a, indices, axis=None, mode='clip')
|
||
|
||
从数组中沿指定轴提取元素。
|
||
当 ``axis`` 不为None时，此函数的功能类似于“fancy”索引（使用数组进行索引）；但如果需要沿给定轴提取元素，它可能更易于使用。例如， ``np.take(arr, indices, axis=3)`` 等价于 ``arr[:,:,:,indices,...]`` 。
|
||
|
||
.. note::
|
||
不支持Numpy的 ``out`` 参数。不支持 ``mode = 'raise'`` ，默认 ``mode`` 为 ``'clip'`` 。
|
||
|
||
参数：
|
||
- **a** (Tensor) - 源数组，shape为 ``(Ni…, M, Nk…)`` 。
|
||
- **indices** (Tensor) - shape为 ``(Nj...)`` 的索引，表示要提取的值。
|
||
- **axis** (int, 可选) - 用于选择值的轴。默认情况下，使用展平后的输入数组。默认值： ``None`` 。
|
||
- **mode** ('raise', 'wrap', 'clip', 可选) - 指定超出范围的索引的行为。默认值： ``'clip'`` 。
|
||
|
||
- 'raise' - 抛出错误；
|
||
- 'wrap' - 循环；
|
||
- 'clip' - 限制在范围内。'clip'模式意味着所有过大的索引都将被替换为该轴最后一个元素的索引。请注意，这将禁止使用负数进行索引。
|
||
|
||
返回：
|
||
Tensor，索引结果。
|
||
|
||
异常：
|
||
- **ValueError** - 如果 ``axis`` 超出范围。
|
||
- **TypeError** - 如果输入不是Tensor。 |