Self Attention Tensorflow Implementation