pytorch/torch/nested
Joel Schlosser 2a8a7425be Fix to wrap jagged dims for split() / split_with_sizes() (#113591)
Still need OpInfo-style tests to catch things like this.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/113591
Approved by: https://github.com/soulitzer
2023-11-14 19:36:08 +00:00
..
_internal Fix to wrap jagged dims for split() / split_with_sizes() (#113591) 2023-11-14 19:36:08 +00:00
__init__.py Implement narrow from a regular tensor to jagged tensor (#112770) 2023-11-13 19:09:59 +00:00