mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
dd618216c5
Summary: We used to only support indexing through - numbers like `x[0, 1]` - tuple like `x[(0, 1)]` - tensor like `x[torch.tensor([0, 1])]` This PR adds support for indexing through list which is equivalent to tensor. - `x[[0, 1, 5]]` - `x[[0, 1], [0, 1]]` - `x[[[0, 1], [0, 1]], [[0, 1], [0, 1]]]` Note for `x[[0, 1, 5]]` we had a bug in AST conversion code so we used to treat it like `x[0, 1, 5]` which means it might accidentally run and produce wrong result(fixes https://github.com/pytorch/pytorch/issues/37286 fixes https://github.com/pytorch/pytorch/issues/18616), now that it's fixed we probably want to mark it as BC breaking. Pull Request resolved: https://github.com/pytorch/pytorch/pull/37848 Reviewed By: suo Differential Revision: D21409840 Pulled By: ailzhang fbshipit-source-id: 6f2d962885c6dc009cb384d98be1822f5ca7a189 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _builtins.py | ||
| _logging.py | ||
| _pickle.py | ||
| _recursive.py | ||
| annotations.py | ||
| frontend.py | ||
| quantized.py | ||
| supported_ops.py | ||
| unsupported_tensor_ops.py | ||