mlx.core.array.argmax

array.argmax(self, axis: Optional[int] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array

参见 argmax()。