cunumeric.nanargmin#

cunumeric.nanargmin(a: ndarray, axis: Any = None, out: ndarray | None = None, *, keepdims: bool = False) → ndarray#

Return the indices of the minimum values in the specified axis ignoring NaNs. For empty arrays, ValueError is raised. For all-NaN slices, ValueError is raised only when CUNUMERIC_NUMPY_COMPATIBILITY environment variable is set, otherwise identity is returned.

Warning: results cannot be trusted if a slice contains only NaNs and -Infs.

Parameters:

a (array_like) – Input array.
axis (int, optional) – By default, the index corresponds to the flattened array, otherwise along the specified axis.
out (ndarray, optional) – If provided, the result will be inserted into this array. It should be of the appropriate shape and dtype.
keepdims (bool, optional) – If this is set to True, the axes which are reduced are left in the result as dimensions with size one. With this option, the result will broadcast correctly against the array.

Returns:

index_array – Array of indices into the array. It has the same shape as a.shape with the dimension along axis removed.

Return type:

ndarray[int]