garage.tf.models.categorical_gru_model moduleΒΆ