I wanted to create a custom Layer in Tensorflow/Keras. I have created an Attention Layer which is called by the following Deepset Layer.
Attention Layer: