cupy.argmax#

cupy.argmax(a, axis=None, dtype=None, out=None, keepdims=False)[source]#

返回沿指定轴的最大值的索引。

参数:

  • a (cupy.ndarray) – 要计算 argmax 的数组。

  • axis (int) – 查找最大值的轴。默认情况下 a 会被展平。

  • dtype – 数据类型指定符。

  • out (cupy.ndarray) – 输出数组。

  • keepdims (bool) – 如果为 True，则将 axis 轴保留为长度为一的轴。

返回:

沿指定轴 a 中最大值的索引。

返回类型:

cupy.ndarray

注意

dtype 和 keepdim 参数是 CuPy 特有的。它们不在 NumPy 中。

注意

axis 参数接受一个整数元组，但这仅限于 CuPy。NumPy 不支持。

另请参阅

numpy.argmax()