pub fn batch_input(w: usize, chunk: &[usize], lines: &[PrepLine]) -> Vec<f32>
Pack one width-batch into the model’s (N, 3, H, W) input buffer.
(N, 3, H, W)