pytorch
22dc22c8 - constraints and tests for masked fill (#80976)

Commit
3 years ago
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
Author
Committer
Parents
Loading