tensorplay.nn.functional.max_pool2d_with_indices

tensorplay.nn.functional.max_pool2d_with_indices(input: TensorBase, kernel_size, stride=None, padding=0, dilation=1, ceil_mode: bool = False, return_indices: bool = True)[source]

Applies a 2D max pooling over an input composed of several input planes, returning (output, indices).

See MaxPool2d for details.