提交 af0972ea authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Correctly broadcast replacement node.

Fix bug reported by John Salvatier
上级 8d046ec8
...@@ -1443,7 +1443,9 @@ def local_upcast_elemwise_constant_inputs(node): ...@@ -1443,7 +1443,9 @@ def local_upcast_elemwise_constant_inputs(node):
# works only for scalars # works only for scalars
cval_i = get_constant_value(i) cval_i = get_constant_value(i)
if all(i.broadcastable): if all(i.broadcastable):
new_inputs.append(T.cast(cval_i, output_dtype)) new_inputs.append(T.shape_padleft(
T.cast(cval_i, output_dtype),
i.ndim))
else: else:
if shape_i is None: if shape_i is None:
return return
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论