maskedtensor
maskedtensor copied to clipboard
Type Promotion for masked operators
🚀 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