opennmt.layers
Module defining reusable and model specific layers.
- Bridge
- ConcatReducer
- CopyBridge
- Dense
- DenseBridge
- DenseReducer
- FeedForwardNetwork
- JoinReducer
- LSTM
- LayerNorm
- LayerWrapper
- MultiHeadAttention
- MultiHeadAttentionReduction
- MultiplyReducer
- PositionEmbedder
- PositionEncoder
- RNN
- RNNCellWrapper
- Reducer
- SelfAttentionDecoderLayer
- SelfAttentionEncoderLayer
- SinusoidalPositionEncoder
- SumReducer
- TransformerLayerWrapper
- ZeroBridge
- combine_heads
- dropout
- future_mask
- gelu
- make_rnn_cell
- split_heads