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

Address the case where sequences are provided as dicts.

上级 72fa916d
...@@ -196,6 +196,8 @@ def scan(fn, ...@@ -196,6 +196,8 @@ def scan(fn,
inner_slices = [] # Actual slices if scan is removed from the picture inner_slices = [] # Actual slices if scan is removed from the picture
# go through sequences picking up time slices as needed # go through sequences picking up time slices as needed
for i, seq in enumerate(seqs): for i, seq in enumerate(seqs):
if isinstance(seq, dict):
seq = seq['input']
actual_slice = seq[0] actual_slice = seq[0]
_seq_val = tensor.as_tensor_variable(seq) _seq_val = tensor.as_tensor_variable(seq)
_seq_val_slice = _seq_val[0] _seq_val_slice = _seq_val[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论