pyvene.models.gru.modelings_gru

Classes

GRUCell(config)
GRUConfig([include_emb, vocab_size, ...])
GRUForClassification(config)
GRULMHeadModel(config)
GRUModel(config)
GRUModelOutput([last_hidden_state, ...])
GRUPreTrainedModel(*inputs, **kwargs)