W3cubDocs

/PyTorch

LocalResponseNorm

class torch.nn.LocalResponseNorm(size: int, alpha: float = 0.0001, beta: float = 0.75, k: float = 1.0) [source]

Applies local response normalization over an input signal composed of several input planes, where channels occupy the second dimension. Applies normalization across channels.

bc=ac(k+αnc=max(0,cn/2)min(N1,c+n/2)ac2)β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}
Parameters
  • size – amount of neighbouring channels used for normalization
  • alpha – multiplicative factor. Default: 0.0001
  • beta – exponent. Default: 0.75
  • k – additive factor. Default: 1
Shape:
  • Input: (N,C,)(N, C, *)
  • Output: (N,C,)(N, C, *) (same shape as input)

Examples:

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

© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.7.0/generated/torch.nn.LocalResponseNorm.html