未验证 提交 c52b6bbf 编写于 作者: F Feiyu Chan 提交者: GitHub

add alias for RNN APIs from paddle.nn.layer.rnn explicitly (#27784)

上级 836ee3b0
......@@ -148,10 +148,17 @@ from .layer.norm import InstanceNorm3d #DEFINE_ALIAS
from .layer.norm import BatchNorm1d #DEFINE_ALIAS
from .layer.norm import BatchNorm2d #DEFINE_ALIAS
from .layer.norm import BatchNorm3d #DEFINE_ALIAS
from .layer.rnn import *
# from .layer.rnn import RNNCell #DEFINE_ALIAS
# from .layer.rnn import GRUCell #DEFINE_ALIAS
# from .layer.rnn import LSTMCell #DEFINE_ALIAS
from .layer.rnn import RNNCellBase #DEFINE_ALIAS
from .layer.rnn import SimpleRNNCell #DEFINE_ALIAS
from .layer.rnn import LSTMCell #DEFINE_ALIAS
from .layer.rnn import GRUCell #DEFINE_ALIAS
from .layer.rnn import RNN #DEFINE_ALIAS
from .layer.rnn import BiRNN #DEFINE_ALIAS
from .layer.rnn import SimpleRNN #DEFINE_ALIAS
from .layer.rnn import LSTM #DEFINE_ALIAS
from .layer.rnn import GRU #DEFINE_ALIAS
from .layer.transformer import MultiHeadAttention
from .layer.transformer import TransformerEncoderLayer
from .layer.transformer import TransformerEncoder
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册