torch.nn.functional.lp_pool2d¶

torch.nn.functional.lp_pool2d(input, norm_type, kernel_size, stride=None, ceil_mode=False)[源代码]¶

对由多个输入平面组成的输入信号应用 2D 幂平均池化。

如果所有输入的 p 次幂之和为零，则梯度也设置为零。