tensorplay.nn.functional.fractional_max_pool3d_with_indices

tensorplay.nn.functional.fractional_max_pool3d_with_indices(input: TensorBase, kernel_size, output_size=None, output_ratio=None, return_indices: bool = True, _random_samples=None)[source]

Applies 3D fractional max pooling over an input signal composed of several input planes, returning (output, indices).

Each plane consumes three random samples ordered (T, H, W), matching aten/src/ATen/native/FractionalMaxPool3d.cpp.