code: def __init__(self, units):
      super(BahdanauAttention, self).__init__()
      self.W1 = tf.keras.layers.Dense(units)
      self.W2 = tf.keras.layers.Dense(units)
      self.V = tf.keras.layers.Dense(1)
  
imports: import tensorflow
  from tensorflow.python.util.tf_export import keras_export
label_desc: Use image node_id 1 to create a new BahdanauAttention object from inherited base classes: tf.keras.layers with example: obj = BahdanauAttention(units)