提交 05e5bf79 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

fixed an error message in scan

上级 1594e248
...@@ -260,12 +260,16 @@ class Scan(PureOp): ...@@ -260,12 +260,16 @@ class Scan(PureOp):
zip(self.inner_seqs(self.inputs), zip(self.inner_seqs(self.inputs),
self.outer_seqs(inputs))): self.outer_seqs(inputs))):
if inner_seq.type.dtype != outer_seq[idx].type.dtype: if inner_seq.type.dtype != outer_seq[idx].type.dtype:
assert isinstance(idx,int)
raise ValueError(err_msg1 % ('sequence', raise ValueError(err_msg1 % ('sequence',
str(outer_seq), str(outer_seq),
idx, idx,
outer_seq.type.dtype, outer_seq.type.dtype,
outer_seq.ndim,
str(inner_seq), str(inner_seq),
inner_seq.type.dtype)) inner_seq.type.dtype,
inner_seq.ndim))
argoffset += len(self.outer_seqs(inputs)) argoffset += len(self.outer_seqs(inputs))
# Check that this 3 things have the same dtype for mit_mot: # Check that this 3 things have the same dtype for mit_mot:
# - initial state of the output # - initial state of the output
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论