fix dynamic decode imperative (#28160)

bc460692 · liu zhengxi · GitHub · 2d45d9a0 · bc460692
显示空白变更内容
内联并排

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