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 axisaxis
is preserved as an axis of length one.
Returns: The indices of the minimum of
a
along an axis.Return type: Note
dtype
andkeepdim
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