Function adaptive_avg_pool2d

pub fn adaptive_avg_pool2d<B>(
    x: Tensor<B, 4>,
    output_size: [usize; 2],
) -> Tensor<B, 4>
where
    B: Backend,

Applies a 2D adaptive avg pooling.