mlx_rs::transforms

Function grad_with_argnums

Source 
pub fn grad_with_argnums<'a, F, Args, Output, Err>(
    f: F,
    argnums: impl IntoOption<&'a [i32]>,
) -> impl FnMut(Args) -> Result<Output> + 'a
where
    F: IntoGrad<'a, Args, Output, Err>,
Expand description

Returns a function which computes the gradient of f.

See also grad for a version that uses the default argument numbers &[0].