mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
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:
committed by
PyTorch MergeBot
parent
2035f6b2e6
commit
7158aa22e8
@ -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]))
|
||||
|
Reference in New Issue
Block a user