RuntimeError: The size of tensor a (127) must match the size of tensor b (40) at non-singleton dimension 3

[shefali1234]

if mask is not None:
mask = mask.unsqueeze(1)
scores = scores.masked_fill(mask == 0, -1e9)
scores = F.softmax(scores, dim=-1)

runtime error in scores.masked_fill(mask == 0, -1e9)
Same for this:
x = x + Variable(self.pe[:,:seq_len], requires_grad=False).cuda()
x = x + pe
return self.dropout(x)

Please help