Function max_pool3d 

pub fn max_pool3d<B: Backend>(
    input: Tensor<B, 5>,
    kernel_size: [usize; 3],
    stride: [usize; 3],
    padding: [usize; 3],
    dilation: [usize; 3],
    ceil_mode: bool,
) -> Tensor<B, 5>

Available on crate feature api only.

Pool native [batch, channels, depth, height, width] activations by maximum.

Executes the existing spatial and depth pooling kernels on the tensor’s backend. Padding, dilation and ceil mode are applied independently per axis. Layout transformations and both pooling stages remain differentiable.