add cudnn lstm; test=develop

add_cudnn_lstm
phlrain 6 years ago
parent 1ffe41d722
commit 084ff6574d

@ -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)

Loading…
Cancel
Save