提交 cd253ea9 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Use value_zeros() instead of always using numpy arrays for initial values in scan.

上级 384337cc
...@@ -778,8 +778,7 @@ class Scan(PureOp): ...@@ -778,8 +778,7 @@ class Scan(PureOp):
# broadcastable dimensions, 0 on the others). # broadcastable dimensions, 0 on the others).
default_shape = [1 if _b else 0 default_shape = [1 if _b else 0
for _b in inp.broadcastable] for _b in inp.broadcastable]
default_val = numpy.zeros(default_shape, default_val = inp.type.value_zeros(default_shape)
dtype=inp.dtype)
wrapped_inp = In(variable=inp, value=default_val, wrapped_inp = In(variable=inp, value=default_val,
update=self.outputs[output_idx]) update=self.outputs[output_idx])
wrapped_inputs.append(wrapped_inp) wrapped_inputs.append(wrapped_inp)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论