局部响应归一化¶
- class torch.nn.LocalResponseNorm(size, alpha=0.0001, beta=0.75, k=1.0)[源代码]¶
对输入信号应用局部响应归一化。
输入信号由多个输入平面组成,其中通道占据第二维。应用跨通道的归一化。
- 参数
- 形状
输入:
输出: (与输入形状相同)
示例
>>> 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)