cupy.argmax#

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

返回沿某个轴的最大值的索引。

参数:

  • 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()