cupy.take

cupy.take(a, indices, axis=None, out=None)

沿指定轴获取数组的指定索引元素。

这是在单个轴上实现“花式索引”的实现。

此函数不支持 mode 选项。

参数:

• a (cupy.ndarray) – 要提取元素的数组。

• indices (int or array-like) – 该函数所接受的元素索引。

• axis (int) – 选择索引的轴。默认使用展平的输入。

• out (cupy.ndarray) – 输出数组。如果提供，它应该具有适当的形状和数据类型。

返回:

花式索引的结果。

返回类型:

cupy.ndarray

参见

numpy.take()