garage.tf.models.categorical_lstm_model moduleΒΆ