mask_logits in layer.py

I think the line "return inputs + mask_value * (1 - mask)" should be "return inputs*mask + mask_value * (1 - mask)"

Open layer.py and locate mask_logits, then inspect how its mask and mask_value arguments are used around the reported return expression. Verify the intended behavior for masked and unmasked positions, and consider the issue resolved when the implementation matches that behavior and the repository's relevant checks pass.

Written by the indexing model from the issue text.