def lstm_bahdanau_attention():
  """Hparams for LSTM with bahdanau attention."""
  hparams = lstm_attention_base()
  hparams.add_hparam("attention_mechanism", "bahdanau")
  return hparams
Hparams for LSTM with bahdanau attention.
tensor2tensor/models/lstm.py