Merge pull request #1106 from Thinkre/patch-1

Laicheng0830 · web-flow · commit 04a08018250b · 2020-10-29T15:58:40.000+08:00
Update recurrent.py
diff --git a/tensorlayer/layers/recurrent.py b/tensorlayer/layers/recurrent.py
@@ -247,7 +247,9 @@ def forward(self, inputs, sequence_length=None, initial_state=None, **kwargs):
                         "but got an actual length of a sequence %d" % i
                     )
 
-            sequence_length = [i - 1 if i >= 1 else 0 for i in sequence_length]
+        sequence_length = tl.layers.retrieve_seq_length_op3(inputs)
+
+        sequence_length = [i - 1 if i >= 1 else 0 for i in sequence_length]
 
         # set warning
         # if (not self.return_last_output) and sequence_length is not None:

Original file line number	Diff line number	Diff line change
`@@ -247,7 +247,9 @@ def forward(self, inputs, sequence_length=None, initial_state=None, **kwargs):`
`247`	`247`	`"but got an actual length of a sequence %d" % i`
`248`	`248`	`)`
`249`	`249`
`250`		`- sequence_length = [i - 1 if i >= 1 else 0 for i in sequence_length]`
	`250`	`+ sequence_length = tl.layers.retrieve_seq_length_op3(inputs)`
	`251`	`+`
	`252`	`+ sequence_length = [i - 1 if i >= 1 else 0 for i in sequence_length]`
`251`	`253`
`252`	`254`	`# set warning`
`253`	`255`	`# if (not self.return_last_output) and sequence_length is not None:`