局部响应归一化¶
- class torch.nn.LocalResponseNorm(size, alpha=0.0001, beta=0.75, k=1.0)[源代码]¶
对输入信号应用局部响应归一化。
输入信号由多个输入平面组成,其中通道占据第二维度。 在通道之间应用归一化。
- Parameters
- Shape:
输入:
输出: (与输入形状相同)
示例:
>>> lrn = nn.LocalResponseNorm(2) >>> signal_2d = torch.randn(32, 5, 24, 24) >>> signal_4d = torch.randn(16, 5, 7, 7, 7, 7) >>> output_2d = lrn(signal_2d) >>> output_4d = lrn(signal_4d)