Function center_crop_2d 

pub fn center_crop_2d(
    input: &ArrayD<f64>,
    crop_h: usize,
    crop_w: usize,
) -> Result<ArrayD<f64>, AugmentationError>

Center crop: crop [crop_h, crop_w] from the center of the last two spatial dims.