Function max_pool2d_with_indices 

pub fn max_pool2d_with_indices(
    x: Tensor<4>,
    options: MaxPoolOptions<2>,
) -> (Tensor<4>, Tensor<4, Int>)

Applies a 2D max pooling with indices.

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