Skip to main content

sparse_hessian_vec

Function sparse_hessian_vec 

Source
pub fn sparse_hessian_vec<F: Float + BtapeThreadLocal, const N: usize>(
    f: impl FnOnce(&[BReverse<F>]) -> BReverse<F>,
    x: &[F],
) -> (F, Vec<F>, SparsityPattern, Vec<F>)
Available on crate feature bytecode only.
Expand description

Batched sparse Hessian: packs N colors per sweep using DualVec.

Like sparse_hessian but reduces sweeps from num_colors to ceil(num_colors / N).