fix rnncell errmsg (#26298)

5 years ago · daad625121
parent 79f12595d8
commit daad625121
1 changed files with 2 additions and 1 deletions
--- 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)):