未验证 提交 7f1e81fd 编写于 作者: A Aurelius84 提交者: GitHub

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

上级 c2c9891b
...@@ -257,7 +257,8 @@ def norm(x, p='fro', axis=None, keepdim=False, name=None): ...@@ -257,7 +257,8 @@ def norm(x, p='fro', axis=None, keepdim=False, name=None):
if in_dygraph_mode(): if in_dygraph_mode():
if dim is None: 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) return _C_ops.final_state_frobenius_norm(input, dim, keepdim, False)
if _in_legacy_dygraph(): if _in_legacy_dygraph():
if dim is None: if dim is None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册