pytorch/torch/nested
Joel Schlosser 6fee208064 Handle -1 in jagged layout NT view ops (#115843)
Allows for inheriting the ragged and batch dims via -1:
```python
nt.view(-1, -1, D)
nt.expand(B, -1, D)
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115843
Approved by: https://github.com/soulitzer
ghstack dependencies: #115636
2023-12-15 00:42:47 +00:00
..
_internal Handle -1 in jagged layout NT view ops (#115843) 2023-12-15 00:42:47 +00:00
__init__.py Implement narrow from a regular tensor to jagged tensor (#112770) 2023-11-13 19:09:59 +00:00