fix rnncell errmsg (#26298) (#26501)

29493ff0 · liu zhengxi · GitHub · b2432b2c · 29493ff0
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

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

未找到文件。
--- a/python/paddle/fluid/layers/rnn.py
+++ b/python/paddle/fluid/layers/rnn.py
@@ -126,7 +126,8 @@ class RNNCell(object):
        else:
            integer_types = (int, )
        check_variable_and_dtype(batch_ref, 'batch_ref',
-                                 ['float32', 'float64'], 'RNNCell')
+                                 ['float32', 'float64', 'int32', 'int64'],
+                                 'RNNCell')
        check_type(shape, 'shape', (list, tuple, type(None), integer_types),
                   'RNNCell')
        if isinstance(shape, (list, tuple)):