tensorplay.nn.functional.lp_pool2d

tensorplay.nn.functional.lp_pool2d(input: Tensor, norm_type, kernel_size, stride=None, ceil_mode: bool = False) → Tensor[source]

Apply a 2D power-average pooling over an input signal.

If the sum of all inputs to the power of p is zero, the gradient is set to zero as well.

See LPPool2d for details.