diff --git a/python/paddle/tensor/linalg.py b/python/paddle/tensor/linalg.py index 2fcf9ff4213d4ec2b176fbca9922d6403de80204..b315e3e9673fc40ae0967b1c7e3f2832827c7f1b 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: