Daniël de Kok
Use Torch attention implementation
a0351e0