fix dynamic decode imperative (#28162)

5d94a5ca · liu zhengxi · GitHub · 4193ae0d · 5d94a5ca
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/layers/rnn.py python/paddle/fluid/layers/rnn.py +1 -1

未找到文件。
--- 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