cupy.argmin#

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

Returns the indices of the minimum along an axis.

Parameters:
  • a (cupy.ndarray) – Array to take argmin.

  • axis (int) – Along which axis to find the minimum. a is flattened by default.

  • dtype – Data type specifier.

  • out (cupy.ndarray) – Output array.

  • keepdims (bool) – If True, the axis axis is preserved as an axis of length one.

Returns:

The indices of the minimum of a along an axis.

Return type:

cupy.ndarray

Note

dtype and keepdim arguments are specific to CuPy. They are not in NumPy.

Note

axis argument accepts a tuple of ints, but this is specific to CuPy. NumPy does not support it.

See also

numpy.argmin()