fix dynamic decode imperative (#28162)

5 years ago · 5d94a5ca54
parent 4193ae0d02
commit 5d94a5ca54
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/rnn.py
+++ b/python/paddle/fluid/layers/rnn.py
@ -1395,7 +1395,7 @@ def _dynamic_decode_imperative(decoder,
        control_flow.increment(x=step_idx_tensor, value=1.0, in_place=True)
        step_idx += 1

-        control_flow.logical_not(nn.reduce_all(finished), cond)
+        cond = control_flow.logical_not(nn.reduce_all(finished))
        if max_step_num is not None and step_idx > max_step_num:
            break