Yuanhao Ji 0a7eef140b Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) ... Fixes #153790 Pull Request resolved: https://github.com/pytorch/pytorch/pull/154022 Approved by: https://github.com/Skylion007		2025-05-27 14:10:00 +00:00
..
__init__.py
nested_int.py	Switch to using Python nested int (#141166)	2024-12-02 19:17:30 +00:00
nested_tensor.py	Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022)	2025-05-27 14:10:00 +00:00
ops.py	Small improvements to NJT matrix multiplies (#146405)	2025-02-06 04:51:12 +00:00
sdpa.py	[cuDNN][SDPA][Nested Tensor] Experimental cuDNN Nested Tensor SDPA Support (forward only) (#141178)	2025-03-04 23:09:09 +00:00