Thanks for the tenosrs.
The uploaded scores already contain NaN values in rows [9, 24, 26], so that the softmax output will also contain NaN values.
Also, the mask contains False for all entries.
Could you post the input tensors and code snippet, which produces these NaNs or are you unsure where these are created?