pytorch/torch/nested
Joel Schlosser 1474eb5f29 Fix jagged composite impl of flatten() (#115192)
Need to handle this in `NestedTensor.__torch_function__()` since it's CompositeImplicit
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115192
Approved by: https://github.com/cpuhrsch, https://github.com/soulitzer
2023-12-19 19:15:21 +00:00
..
_internal Fix jagged composite impl of flatten() (#115192) 2023-12-19 19:15:21 +00:00
__init__.py Implement narrow from a regular tensor to jagged tensor (#112770) 2023-11-13 19:09:59 +00:00