pytorch/torch/nested
Joel Schlosser 3e2f276a14 Fix to() on non-contiguous NJTs (#137124)
Called out via torchrec integration: `lengths` is not handled properly.

Future work (not related to non-contiguous NJTs): #137275
Pull Request resolved: https://github.com/pytorch/pytorch/pull/137124
Approved by: https://github.com/soulitzer
ghstack dependencies: #137030, #137031
2024-10-08 15:11:05 +00:00
..
_internal Fix to() on non-contiguous NJTs (#137124) 2024-10-08 15:11:05 +00:00
__init__.py Implement 2D version of masked_select for nestedtensors (#133889) 2024-08-20 21:46:32 +00:00