Function predict_into 

pub fn predict_into(
    mlp: &Mlp,
    input: &[f32],
    scratch: &mut Scratch,
    out: &mut [f32],
) -> Result<()>

Shape-safe, non-allocating inference.

Thin wrapper around Mlp::predict_into.