提交 084ff657 编写于 作者: P phlrain

add cudnn lstm; test=develop

上级 1ffe41d7
...@@ -525,6 +525,9 @@ def cudnn_lstm(input, ...@@ -525,6 +525,9 @@ def cudnn_lstm(input,
default_initializer(Initialize|None): Where use initializer to initialize the Weight default_initializer(Initialize|None): Where use initializer to initialize the Weight
If set None, defaule initializer will be used 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: Returns:
rnn_out(Tensor): result of LSTM hidden, shape is (seq_len x batch_size x hidden_size) rnn_out(Tensor): result of LSTM hidden, shape is (seq_len x batch_size x hidden_size)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册