mlx.core.argmin

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

沿轴的最小值的索引。

参数:

• a (array) – 输入数组。

• axis (int, 可选) – 可选的归约轴。如果未指定，默认为对整个数组进行归约。

• keepdims (bool, 可选) – 保持被归约的轴作为单维度，默认为 False。

返回:

一个包含最小值索引的 uint32 数组。

返回类型:

array