提交 307a28ff authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fixed bug in make_thunk

I used the same variable name for two different variables, which resulted into a bug.
上级 0c358fb1
...@@ -194,7 +194,7 @@ class ScanOp(PureOp): ...@@ -194,7 +194,7 @@ class ScanOp(PureOp):
base_inputs = self.inputs[:len(self.outputs)] base_inputs = self.inputs[:len(self.outputs)]
base_buffers = node_input_storage[1: 1 + len(base_inputs)] base_buffers = node_input_storage[1: 1 + len(base_inputs)]
aux_inputs = self.inputs[len(self.outputs):] aux_inputs = self.inputs[len(self.outputs):]
aux_buffers = node_input_storage[1 + len(base_inputs):] aux_membuffers = node_input_storage[1 + len(base_inputs):]
# 2.1 First the auxiliary arguments, those that are parameters or # 2.1 First the auxiliary arguments, those that are parameters or
# input # input
...@@ -208,8 +208,7 @@ class ScanOp(PureOp): ...@@ -208,8 +208,7 @@ class ScanOp(PureOp):
non_tensor_args = [] non_tensor_args = []
non_tensor_buffers = [] non_tensor_buffers = []
aux_buffers = [] aux_buffers = []
for mem_buf, var in izip(aux_buffers, aux_inputs): for mem_buf, var in izip(aux_membuffers, aux_inputs):
if mem_buf[0] is not None: if mem_buf[0] is not None:
givens[var] = theano.shared(mem_buf[0], name=var.name, givens[var] = theano.shared(mem_buf[0], name=var.name,
borrow=True) borrow=True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论