class_docstr: TODO: implement forward_attention
  TODO: location sensitive attention
  TODO: implement attention windowing 
class_docstr_tok: ["TODO",":","implement","forward_attention","TODO",":","location","sensitive","attention","TODO",":","implement","attention","windowing"]
code: def update_states(
      self, old_states, scores_norm, attn_weights, new_alpha=None
  ):
      states = []
      if self.use_loc_attn:
          states = [old_states[0] + scores_norm, attn_weights]
      if self.use_forward_attn:
          states.append(new_alpha)
      return tuple(states)
  
imports: import tensorflow
  from tensorflow import keras
  from tensorflow.python.ops import math_ops
label_desc: Use image node_id 8 for calling the Attention obj's underlying member method code with example usage: obj.update_states(old_states, scores_norm, attn_weights, new_alpha) and returns: tuple