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.