pytorch / maskedtensor

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

binary op sparse fix #59

Closed george-qi closed 2 years ago

george-qi commented 2 years ago

Stack from ghstack:

Differential Revision: D36109702

Fixing an issue where the entire last row of mask could be False, therefore resulting in a sparse tensor that doesn't inherit the original size and becomes smaller than intended

george-qi commented 2 years ago

@george-qi has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.