提交 3e28e55e authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Minor typo fix

上级 c56b57f3
......@@ -2185,7 +2185,7 @@ class MaxAndArgmax(Op):
x = _as_tensor_variable(x)
if isinstance(axis, Variable):
if not isinstance(axis, Constant):
raise TypeError("MaxAndArgmax need a constant axis")
raise TypeError("MaxAndArgmax needs a constant axis")
axis = [axis.data]
if isinstance(axis, int):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论