attention block for keras Functional Model with only tensorflow backend - View it on GitHub