pytorch

85258ec1 - Add mask_type=2 to masked_softmax for when mask.size() == input.size() (#85915)

Add mask_type=2 to masked_softmax for when mask.size() == input.size() (#85915) Pull Request resolved: https://github.com/pytorch/pytorch/pull/85915 Approved by: https://github.com/cpuhrsch