fix paddle.var() stop_gradient error (#50063)

9737525a · sneaxiy · GitHub · 964cd660 · 9737525a
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/tensor/stat.py python/paddle/tensor/stat.py +1 -0

未找到文件。
--- a/python/paddle/tensor/stat.py
+++ b/python/paddle/tensor/stat.py
@@ -158,6 +158,7 @@ def var(x, axis=None, unbiased=True, keepdim=False, name=None):
    if unbiased:
        one_const = paddle.ones([], x.dtype)
        n = where(n > one_const, n - 1.0, one_const)
+    n.stop_gradient = True
    out /= n
    return out