rai_core::utils

Function numerical_jvp

Source 
pub fn numerical_jvp<F>(
    func: F,
    input: impl AsRef<Tensor>,
    tangent: impl AsRef<Tensor>,
    eps: f64,
) -> Tensor
where
    F: for<'a> Func<Basic, &'a Tensor, Tensor>,