mlx.core.argmax

目录

mlx.core.argmax#

argmax(a: array, /, axis: None | int = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

Indices of the maximum values along the axis.

参数:

a (array) -- Input array.
axis (int, optional) -- Optional axis to reduce over. If unspecified this defaults to reducing over the entire array.
keepdims (bool, optional) -- Keep reduced axes as singleton dimensions, defaults to False.

返回:

The uint32 array with the indices of the maximum values.

返回类型: