Skip to main content

max_pool1d_with_indices

Function max_pool1d_with_indices 

pub fn max_pool1d_with_indices(
    x: Tensor<3>,
    options: MaxPoolOptions<1>,
) -> (Tensor<3>, Tensor<3, Int>)
Expand description

Applies a 1D max pooling with indices.

Supports symmetric and asymmetric padding through MaxPoolOptions. Returned indices always refer to positions in the unpadded input.