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)