attention mechanisms with tensorflow