提交 33bda7ca authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3734 from carriepl/cudandarray_data

Fix index error in scan backend
......@@ -1323,7 +1323,7 @@ class Scan(PureOp):
if var is None:
old_mitmot_input_data[idx] = None
elif self.inps_is_tensor[idx]:
elif self.inps_is_tensor[idx + self.n_seqs]:
old_mitmot_input_data[idx] = var.data
else:
old_mitmot_input_data[idx] = var.gpudata
......
This source diff could not be displayed because it is too large. You can view the blob instead.
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论