remove more (#164753)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164753
Approved by: https://github.com/aorenste, https://github.com/mlazos
ghstack dependencies: #164664, #164665, #164667, #164668
This commit is contained in:
Laith Sakka
2025-10-07 13:13:34 -07:00
committed by PyTorch MergeBot
parent 2035f6b2e6
commit 7158aa22e8
13 changed files with 19 additions and 59 deletions

View File

@ -1453,7 +1453,7 @@ def tensor_split_tensor_indices_or_sections_py_impl(
# To avoid PendingUnbackedSymbolNotFound errors, we tell the compiler it's fine to not bind these.
with ctx():
indices = [i.item() for i in tensor_indices_or_sections]
# WARNING: Tempted to torch._check_is_size on the indices here? You
# WARNING: Tempted to torch._check(x>0) on the indices here? You
# can't: tensor_split works with negative values in indices:
#
# >>> torch.tensor_split(torch.randn(10), torch.tensor([-5, 5]))