hookeai.model_architectures.rnn_base_model.custom.gru_vmap

Multi-layer gated recurrent unit (GRU) recurrent neural network model.

Custom implementation of GRUCell and GRU compatible with VMAPs.

Classes

GRUCell(torch.nn.Module)

Gated Recurrent Unit (GRU) cell.

GRU(torch.nn.Module)

Multi-layer gated recurrent unit (GRU) recurrent neural network model.

Classes

GRU(input_size, hidden_size[, num_layers, ...])	Multi-layer gated recurrent unit (GRU) recurrent neural network model.
GRUCell(input_size, hidden_size[, bias, device])	Gated Recurrent Unit (GRU) cell.