mask_logits function

[BangLiu]

I think in mask_logits, maybe "return inputs + (-1e30) * (1 - mask)" shall be replaced by "return inputs * mask + (-1e30) * (1 - mask)".

[andy840314]

I don't think it will make a big difference since -1e30 is already a extremely small value and should mask correctly.