提交 9c4cd7dc authored 作者: Frederic's avatar Frederic

Add comment following code review.

上级 fff84f9f
......@@ -44,6 +44,8 @@ def local_max_and_argmax(node):
"""
if node.op == T._max_and_argmax:
if len(node.outputs[1].clients) == 0:
#MaxAndArgmax support variable axis,
#but CAReduce support only constant axis.
try:
axis = get_scalar_constant_value(node.inputs[1])
except NotScalarConstantError:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论