I am trying to insert an attention mechanism to my simple bio-tagging (multi-class) model. Without attention, the model is working but when tried to insert the attention lay
