Python PyTorch LocalResponseNorm用法及代码示例

本文简要介绍python语言中 torch.nn.LocalResponseNorm 的用法。

用法:

class torch.nn.LocalResponseNorm(size, alpha=0.0001, beta=0.75, k=1.0)

参数：

• size-用于归一化的相邻通道的数量

• alpha-乘法因子。默认值：0.0001

• beta-指数。默认值：0.75

• k-附加因子。默认值：1

对由多个输入平面组成的输入信号应用局部响应归一化，其中通道占据第二维。跨渠道应用标准化。

b_{c} = a_{c}\left(k + \frac{\alpha}{n} \sum_{c'=\max(0, c-n/2)}^{\min(N-1,c+n/2)}a_{c'}^2\right)^{-\beta}

形状：

• 输入：(N, C, *)

• 输出：(N, C, *)(与输入的形状相同)

例子：

>>> 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)