提交 496ba4ba authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Ensure integer axes

上级 daaba020
......@@ -1355,7 +1355,7 @@ class MaxAndArgmax(Op):
if axes is None:
axes = tuple(range(x.ndim))
else:
axes = tuple(axes)
axes = tuple(int(ax) for ax in axes)
max[0] = theano._asarray(numpy.max(x, axes),
dtype=node.outputs[0].dtype)
# Numpy does not support multiple axes for argmax
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论