search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.46k stars 467 forks source link

add `masked.amin` .. `masked.mean` #8014

Closed ManfeiBai closed 1 week ago

ManfeiBai commented 2 weeks ago

add masked.log_softmax .. masked.mean

fix https://github.com/pytorch/xla/issues/7509 fix https://github.com/pytorch/xla/issues/7508