提交 1295dbb1 authored 作者: Sina Honari's avatar Sina Honari

removing a bug

上级 ad8a1755
......@@ -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.prod(values) == 0:
if numpy.any(values == 0):
# All non-broadcasted dimensions should be zero
assert max(values) <= 1
out_shape.append(0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论