提交 f1ef572d authored 作者: SinaHonari's avatar SinaHonari

Merge pull request #1 from lamblin/fix_elemwise_perform

Fix check for 0 in values
......@@ -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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论