fix rnncell errmsg (#26298)

daad6251 · liu zhengxi · GitHub · 79f12595 · daad6251
隐藏空白更改
内联并排

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
@@ -127,7 +127,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)):