I uploaded the two tensors:
Scores: https://easyupload.io/6j6lvd 3
Masks: https://easyupload.io/vixfm9 1
This is the warning I am getting: WARNING:root:NaN or Inf found in input tensor.
Thanks for the help!