mindspore.numpy.take

mindspore.numpy.take(a, indices, axis=None, mode='clip')

从数组中沿指定轴提取元素。 当 axis 不为None时，此函数的功能类似于“fancy”索引（使用数组进行索引）；但如果需要沿给定轴提取元素，它可能更易于使用。例如， np.take(arr, indices, axis=3) 等价于 arr[:,:,:,indices,...] 。

说明

不支持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。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> a = np.array([4, 3, 5, 7, 6, 8])
>>> indices = np.array([0, 1, 4])
>>> output = np.take(a, indices)
>>> print(output)
[4 3 6]
>>> indices = np.array([[0, 1], [2, 3]])
>>> output = np.take(a, indices)
>>> print(output)
[[4 3]
[5 7]]