Function adaptive_avg_pool2d
pub fn adaptive_avg_pool2d(x: Tensor<4>, output_size: [usize; 2]) -> Tensor<4>Expand description
Applies a 2D adaptive avg pooling.
pub fn adaptive_avg_pool2d(x: Tensor<4>, output_size: [usize; 2]) -> Tensor<4>Applies a 2D adaptive avg pooling.