pytorch

cb687773 - masked argmin and argmax

masked argmin and argmax Pull Request resolved: https://github.com/pytorch/pytorch/pull/74525 Approved by: https://github.com/cpuhrsch