From bb704f8f6c2f41c6aab8f8c5bdf84699fd3337ac Mon Sep 17 00:00:00 2001 From: jyz Date: Fri, 26 Jun 2026 01:13:26 +0000 Subject: [PATCH] Fix #1855: LocalResponseNorm +/-Inf --- torch/nn/modules/normalization.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/torch/nn/modules/normalization.py b/torch/nn/modules/normalization.py index c4df21528cae2..59a0c611ae868 100644 --- a/torch/nn/modules/normalization.py +++ b/torch/nn/modules/normalization.py @@ -57,6 +57,14 @@ def __init__( self.alpha = alpha self.beta = beta self.k = k + if k <= 0 and alpha <= 0 and beta != 0: + raise ValueError( + f"LocalResponseNorm: k and alpha cannot both be non-positive " + f"when beta is non-zero; " + f"got k={k}, alpha={alpha}, beta={beta}. " + f"The denominator k + (alpha/size) * sum(x^2) would collapse to zero " + f"and 0^(-beta) = inf." + ) def forward(self, input: Tensor) -> Tensor: """