burn::tensor::module

Function adaptive_avg_pool1d

pub fn adaptive_avg_pool1d<B>(
    x: Tensor<B, 3>,
    output_size: usize,
) -> Tensor<B, 3>
where
    B: Backend,
Expand description

Applies a 1D adaptive avg pooling.