From 45e43dfe36dbb43394b6b72f5959d2a0cfa4ba90 Mon Sep 17 00:00:00 2001 From: Aurelius84 Date: Wed, 13 Apr 2022 09:54:49 +0800 Subject: [PATCH] [Eager]Fix final_state_frobenius_norm BUG with axis==None (#41693) --- python/paddle/tensor/linalg.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/tensor/linalg.py b/python/paddle/tensor/linalg.py index 2fcf9ff4213..b315e3e9673 100644 --- a/python/paddle/tensor/linalg.py +++ b/python/paddle/tensor/linalg.py @@ -347,7 +347,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: -- GitLab