How can i change the attention translation example in pytorch to use hard attention?

i want to use Hard attention instead of soft attention in the translation example :blush: