masking in EPiC attention layer

[dfaroughy]

this does not seem to work. It produces a bunch of nans in the output..