提交 8e36e584 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix check for 0 in values

上级 2454257b
......@@ -821,7 +821,7 @@ class Elemwise(Op):
# Determine the shape of outputs
out_shape = []
for values in izip(*[input.shape for input in inputs]):
if numpy.any(values == 0):
if any(v == 0 for v in values):
# All non-broadcasted dimensions should be zero
assert max(values) <= 1
out_shape.append(0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论