I\'m trying to build an encoder-decoder model using attention mechanism with tensorflow. I\'m using tensorflow_addons repository trying to reproduce and understand this mode