diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index ef46bbb2717a0216df27337424f1211b5689e5d6..d2df819c0853ccd0f12976ac06c9813471320bc1 100644 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -525,6 +525,9 @@ def cudnn_lstm(input, default_initializer(Initialize|None): Where use initializer to initialize the Weight If set None, defaule initializer will be used + fix_seed(bool): If it's True, fix seed will used for dropout in LSTM + seed(int): If fix_seed is True, dropout seed in LSTM will use this seed + Returns: rnn_out(Tensor): result of LSTM hidden, shape is (seq_len x batch_size x hidden_size)