pytorch/torch/nested
Joel Schlosser 17c9c70265 Support for torch.nested.as_nested_tensor(t) (#113280)
This PR adds support for tensor inputs to `as_nested_tensor()`. The tensor is treated as a batch of consistently-sized constituents. It utilizes `_nested_view_from_values_offsets()` to return a real view that allows for propagating gradients into inputs.
Co-authored-by: voznesenskym <voznesenskym@gmail.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/113280
Approved by: https://github.com/cpuhrsch, https://github.com/soulitzer
2024-03-21 04:13:55 +00:00
..
_internal Proper view support for jagged layout NestedTensor (#113279) 2024-03-20 23:45:34 +00:00
__init__.py Support for torch.nested.as_nested_tensor(t) (#113280) 2024-03-21 04:13:55 +00:00