cupy.take#

cupy.take(a, indices, axis=None, out=None)[source]#

沿指定轴取出数组中特定索引位置的元素。

这是沿单轴实现“花式索引”。

此函数不支持 mode 选项。

参数:

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

  • indices (int 或 类数组) – 此函数提取的元素的索引。

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

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

返回:

花式索引的结果。

返回类型:

cupy.ndarray

另请参见

numpy.take()