pytorch

54a1cc52 - Support softmax(nested tensor) (#80179)

Support softmax(nested tensor) (#80179) Pull Request resolved: https://github.com/pytorch/pytorch/pull/80179 Approved by: https://github.com/cpuhrsch, https://github.com/jbschlosser