11 Commits

Author	SHA1	Message	Date
Mikayla Gawarecki	ec61951f07	Fix inaccuracy in nt constructor documentation + broken rendering (#89152) ... Rendering was broken and docstring seemed to be inaccurate  Pull Request resolved: https://github.com/pytorch/pytorch/pull/89152 Approved by: https://github.com/cpuhrsch	2022-11-16 22:32:46 +00:00
Antoni Viros i Martin	c77368d416	Implement a constructor for nested_tensor that is similar to torch.tensor() (#88213) ... Summary: This diff merges both previous implementations of constructors for nested tensors, the one from lists of tensors and the one with arbitrary python lists, adn implements it in pytorch core so no extensions are needed to construct NT. Pull Request resolved: https://github.com/pytorch/pytorch/pull/88213 Approved by: https://github.com/cpuhrsch	2022-11-08 00:03:18 +00:00
Frankie Robertson	afc9963865	Fix path to nested_tensor in example (#86891) ... This appears to be a typo. Pull Request resolved: https://github.com/pytorch/pytorch/pull/86891 Approved by: https://github.com/H-Huang	2022-10-13 17:42:32 +00:00
Mikayla Gawarecki	a77f2a95a7	Improve NestedTensor documentation (#85186) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/85186 Approved by: https://github.com/cpuhrsch	2022-10-12 22:03:04 +00:00
Mikayla Gawarecki	afaee00fec	Add python nested_tensor and as_nested_tensor constructors in torch.nested (#85593) ... Remove `torch.nested_tensor` which has erroneous behavior wrt gradients (could be either leaf or not leaf). Introduce `torch.nested.nested_tensor` and `torch.nested.as_nested_tensor` in the vein of `torch.tensor` and `torch.as_tensor`. Done in nested `__init__.py` for now but can move to pybind in future (when we want to load from numpy/nested lists ). Discussed offline with @cpuhrsch and pybind constructor (https://github.com/pytorch/pytorch/pull/85536) was more gnarly than expected, so we can move to that when we do need loading from numpy etc. Differential Revision: [D39806622](https://our.internmc.facebook.com/intern/diff/D39806622) Pull Request resolved: https://github.com/pytorch/pytorch/pull/85593 Approved by: https://github.com/drisspg, https://github.com/cpuhrsch	2022-09-28 20:15:02 +00:00
Mikayla Gawarecki	e217b30b0f	Add torch.nested namespace (#84102) ... First step towards #83775 - only `to_padded_tensor` is moved to the nested namespace for now - following the schema used for `special`, `fft`, `linalg` and other namespaces, nested functions are registered in native_functions.yaml as `nested_{function_name}` and are bound to the desired Python name in `torch/nested/__init__.py`, and the desired C++ name in `torch/csrc/api/include/torch/nested.h`. ~~**Question**: should we keep the documentation for `Tensor.to_padded_tensor` or can this deleted since it is shared by `torch.nested.to_padded_tensor`?~~ [generated nested docs](https://docs-preview.pytorch.org/84102/nested.html?highlight=nested#module-torch.nested) Differential Revision: [D39361148](https://our.internmc.facebook.com/intern/diff/D39361148) Pull Request resolved: https://github.com/pytorch/pytorch/pull/84102 Approved by: https://github.com/drisspg	2022-09-12 16:31:05 +00:00
Scott Wolchok	f9d0bc5338	[PyTorch] Delete NestedTensor Python wrapper (#74691) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74691 The wrapper just called through to methods on the underlying Tensor. ghstack-source-id: 152433754 Test Plan: existing tests Reviewed By: ezyang Differential Revision: D34689789 fbshipit-source-id: cf53476780cf3ed00a3aa4add441300bfe8e27ce (cherry picked from commit 5a9e5eb6bc13eb30be6e3c3bc4ac954c92704198)	2022-03-29 19:13:40 +00:00
Scott Wolchok	f7fe75c766	[PyTorch] Call self._impl.unbind directly in _nestedtensor wrapper (#74001) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74001 Incremental step toward removing this wrapper. ghstack-source-id: 151540965 Test Plan: existing test_nestedtensor Reviewed By: ezyang Differential Revision: D34689382 fbshipit-source-id: 46eb203d4fc9ccc30008c7dc3ada24cab47b57d4 (cherry picked from commit 30e204362ca6a3da05e6d99726489879c9253a0e)	2022-03-17 18:04:50 +00:00
Scott Wolchok	79c3bbceb2	[PyTorch] Move NestedTensor printing to _tensor_str.py (#74000) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74000 Now that we're in-core, we can just customize this. ghstack-source-id: 151540966 Test Plan: Existing test_nestedtensor seems to pass Reviewed By: ezyang Differential Revision: D34665270 fbshipit-source-id: 5097944a4dc4fe80cea2b8576f0123466dbeab43 (cherry picked from commit d0315f46f9906c904639f43f218e439407f5b2a7)	2022-03-17 18:04:50 +00:00
Scott Wolchok	88b19da49a	[PyTorch] Make NestedTensor::dim() work (#73679) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/73679 We can update the TensorImpl state used to track dim() just fine. I'm not sure if this is sustainable; do we *want* callers to be able to muck with nested_size_tensor_ directly? ghstack-source-id: 150349610 Test Plan: Updated test_nestedtensor. Reviewed By: cpuhrsch Differential Revision: D34570523 fbshipit-source-id: 739555d63226f925d6a502c9c742ce5f431cb6cc (cherry picked from commit 1bb188162f3639f26a6204ad5d40f73e4c664a6d)	2022-03-04 19:35:41 +00:00
Christian Puhrsch	484c0de670	Minimal NestedTensor (#72881) ... Summary: This PR adds a minimal version of a NestedTensor. It introduces the general harness future development can be built around. Pull Request resolved: https://github.com/pytorch/pytorch/pull/72881 Reviewed By: albanD Differential Revision: D34259177 Pulled By: cpuhrsch fbshipit-source-id: 0245c36f603424e20f3b09651043c207f526d760 (cherry picked from commit 10764e8d427f29b364567e4cbc86ed73c3933158)	2022-03-02 16:31:51 +00:00