rnn

Implementations of flexible GRU and LSTM that can handle sequences of length 0.

Functions

get_rnn(cell_type)

Get LSTM or GRU.

Classes

GRU(*args, **kwargs)	GRU that can handle zero-length sequences
LSTM(*args, **kwargs)	LSTM that can handle zero-length sequences
RNN(mode, input_size, hidden_size, …)	Base class flexible RNNs.

TimeDistributedEmbeddingBag pytorch_forecasting.models.nn.rnn.get_rnn