[Eager]Fix final_state_frobenius_norm BUG with axis==None (#41693) (#41725)

7f1e81fd · Aurelius84 · GitHub · c2c9891b · 7f1e81fd
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

python/paddle/tensor/linalg.py python/paddle/tensor/linalg.py +2 -1

未找到文件。
--- a/python/paddle/tensor/linalg.py
+++ b/python/paddle/tensor/linalg.py
@@ -257,7 +257,8 @@ def norm(x, p='fro', axis=None, keepdim=False, name=None):

        if in_dygraph_mode():
            if dim is None:
-                return _C_ops.final_state_frobenius_norm(input, keepdim, True)
+                return _C_ops.final_state_frobenius_norm(input, [], keepdim,
+                                                         True)
            return _C_ops.final_state_frobenius_norm(input, dim, keepdim, False)
        if _in_legacy_dygraph():
            if dim is None: