Package ai.djl.nn.recurrent

package ai.djl.nn.recurrent

Contains classes for recurrent neural network operations.

See Also:

• RecurrentBlock

Related Packages

Package	Description
ai.djl.nn	Contains classes to construct neural networks.
ai.djl.nn.convolutional	Contains classes that define convolutional operations extending Convolution and Deconvolution.
ai.djl.nn.core	Contains classes that define simple neural network operations.
ai.djl.nn.norm	Contains classes that define normalizing neural network operations.
ai.djl.nn.pooling	Contains pooling neural network operations in Pool and helpers for it.
ai.djl.nn.transformer	Contains blocks for transformer models.

Class	Description
GRU	GRU is an abstract implementation of recurrent neural networks which applies GRU (Gated Recurrent Unit) recurrent layer to input.
GRU.Builder	The Builder to construct a GRU type of Block.
LSTM	LSTM is an implementation of recurrent neural networks which applies Long Short-Term Memory recurrent layer to input.
LSTM.Builder	The Builder to construct a LSTM type of Block.
RecurrentBlock	RecurrentBlock is an abstract implementation of recurrent neural networks.
RecurrentBlock.BaseBuilder<T extends RecurrentBlock.BaseBuilder>	The Builder to construct a RecurrentBlock type of Block.
RNN	RNN is an implementation of recurrent neural networks which applies a single-gate recurrent layer to input.
RNN.Activation	An enum that enumerates the type of activation.
RNN.Builder	The Builder to construct a RNN type of Block.