pytorch / maskedtensor

MaskedTensors for PyTorch
https://pytorch.org/maskedtensor
Other
38 stars 10 forks source link

Type Promotion for masked operators #57

Open george-qi opened 2 years ago

george-qi commented 2 years ago

🚀 The feature, motivation and pitch

As seen in Issue #54, type promotion has not been implemented for masked reductions. We'd like to implement this to be consistent with PyTorch.

Alternatives

No response

Additional context

No response