Package ai.djl.nn.recurrent

Contains classes for recurrent neural network operations.

See: Description

Class Summary

Class	Description
GRU	Applies GRU (Gated Recurrent Unit) recurrent layer to input.
GRU.Builder	The Builder to construct a GRU type of Block.
LSTM	Applies Long Short-Term Memory recurrent layer to input.
LSTM.Builder	The Builder to construct a LSTM type of Block.
RecurrentCell	Applies recurrent layers to input data.
RecurrentCell.BaseBuilder<T extends RecurrentCell.BaseBuilder>	The Builder to construct a RecurrentCell type of Block.
RNN	Applies a single-gate recurrent layer to input.
RNN.Builder	The Builder to construct a RNN type of Block.

Enum Summary

Enum	Description
RNN.Activation	An enum that enumerates the type of activation.

Package ai.djl.nn.recurrent Description

Contains classes for recurrent neural network operations.

See Also:

RecurrentCell