An implementation of Attention Mechanism

Hi!

Tensorflow provides a implementation of attention mechanism in tensorflow.contrib.seq2seq. I’m just wondering if pytorch also provide an implementation of attention mechanism to start from?
if not, I wish it could be included as nn.Attention

I’m want to use attention mechanism in an image classifier

Thank you