constraints and tests for masked fill (#80976)
For masked_fill we consider similar constraints to addition including broadcasting. This PR covers the case when the argument is a tensor. Therefore, we can consider broadcasting with respect to the underlying tensor to get similar constraints to addition.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80976
Approved by: https://github.com/jamesr66a