Function calculate_pool_output_shape 

pub fn calculate_pool_output_shape<const N: usize>(
    in_shape: &Shape,
    kernel_size: &[usize; N],
    stride: &[usize; N],
    padding: &[usize; N],
    dilation: &[usize; N],
    ceil_mode: bool,
) -> Result<Shape, ShapeError>

Calculate the expected output shape [batch_size, channels_out, spatial_dims, ..] for a pooling operation.