Function predict_logits 

fn predict_logits(
    data: &[f32],
    dims: [usize; 3],
    model: &OnnxModel,
    patch: [usize; 3],
    params: &HdBetParams,
    progress: impl FnMut(usize, usize),
) -> Result<Vec<f32>, OnnxError>

Gaussian-weighted sliding-window logits (2 channels, C-order (z, y, x), shape dims).