mlx_rs::ops

Function divmod

Source
pub fn divmod(
    a: impl AsRef<Array>,
    b: impl AsRef<Array>,
) -> Result<(Array, Array)>
Expand description

Element-wise quotient and remainder.

The fuction divmod(a, b) is equivalent to but faster than (a // b, a % b). The function uses numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

Returns Ok((quotient, remainder)) if the operation was successful.