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.