Attention correction
Merge pull request #53 from laurent-dinh/fix_attention

Attention correction