提交 db0ab00a 编写于 作者: M Megvii Engine Team

fix(imperative/functional): fix general-norm assert always be true

GitOrigin-RevId: f8d4da570f66f36f3c743e67a1a1c1222f33aad8
上级 4fcf8b49
......@@ -1188,7 +1188,7 @@ def general_norm(
sorted(normalized_axis), normalized_axis
)
assert (
normalized_axis[-1] < inp.ndim,
normalized_axis[-1] < inp.ndim
), "the maximum axis in normalized_axis is greater than inp_shape.ndim"
assert len(set(normalized_axis)) == len(
normalized_axis
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册