未验证 提交 45e43dfe 编写于 作者: A Aurelius84 提交者: GitHub

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

上级 acd08a9b
...@@ -347,7 +347,8 @@ def norm(x, p='fro', axis=None, keepdim=False, name=None): ...@@ -347,7 +347,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.
先完成此消息的编辑!
想要评论请 注册