Function conv2d_batch 

pub fn conv2d_batch(
    input: &[f32],
    weight: &[f32],
    bias: &[f32],
    output: &mut [f32],
    batch_size: usize,
    in_channels: usize,
    out_channels: usize,
    input_height: usize,
    input_width: usize,
    kernel_height: usize,
    kernel_width: usize,
    stride_h: usize,
    stride_w: usize,
    padding_h: usize,
    padding_w: usize,
)

2D convolution for batched images Input shape: [batch_size, in_channels, height, width] Weight shape: [out_channels, in_channels, kernel_height, kernel_width] Output shape: [batch_size, out_channels, out_height, out_width]