Skip to main content

max_pool2d_with_indices_backward

Function max_pool2d_with_indices_backward 

pub fn max_pool2d_with_indices_backward(
    x: Tensor<4>,
    kernel_size: [usize; 2],
    stride: [usize; 2],
    padding: [usize; 2],
    dilation: [usize; 2],
    ceil_mode: bool,
    output_grad: Tensor<4>,
    indices: Tensor<4, Int>,
) -> Tensor<4>
Expand description

Backward pass for the max pooling 2d operation.