Attention layer tensorflow. .