Self Attention Pytorch