Function max_pool3d 

pub fn max_pool3d(
    input: &Tensor,
    kernel_size: (usize, usize, usize),
    stride: Option<(usize, usize, usize)>,
    padding: (usize, usize, usize),
    dilation: (usize, usize, usize),
    ceil_mode: bool,
) -> Result<Tensor>

Max pooling 3D