Merge branch 'master' into eig-svd-vals

Jutho · Jun 23, 2024 · 0ed5995 · 0ed5995
2 parents e4d8500 + b026cf2
commit 0ed5995
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ext/TensorKitChainRulesCoreExt.jl b/ext/TensorKitChainRulesCoreExt.jl
@@ -172,7 +172,9 @@ end
 function ChainRulesCore.rrule(::typeof(norm), a::AbstractTensorMap, p::Real=2)
     p == 2 || error("currently only implemented for p = 2")
     n = norm(a, p)
-    norm_pullback(Δn) = NoTangent(), a * (Δn' + Δn) / (n * 2), NoTangent()
+    function norm_pullback(Δn)
+        return NoTangent(), a * (Δn' + Δn) / 2 / hypot(n, eps(one(n))), NoTangent()
+    end
     return n, norm_pullback
 end