cupy.take_along_axis#

cupy.take_along_axis(a, indices, axis)[源代码]#

通过匹配 1D 索引和数据切片从输入数组中获取值。

参数:

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

  • indices (cupy.ndarray) – 沿着 a 的每个 1D 切片要获取的索引。

  • axis (int) – 沿着哪个轴进行 1D 切片。

返回:

索引后的结果。

返回类型:

cupy.ndarray

另请参阅

numpy.take_along_axis()