pytorch/test/cpp
Will Feng b6cee03e29 C++ tensor indexing: add Slice / TensorIndex (#30424)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30424

`at::indexing::TensorIndex` is used for converting C++ tensor indices such as `{None, "...", Ellipsis, 0, true, {1, None, 2}, torch::tensor({1, 2})}` into its equivalent `std::vector<TensorIndex>`, so that further tensor indexing operations can be performed using the supplied indices.

Test Plan: Imported from OSS

Differential Revision: D18695902

Pulled By: yf225

fbshipit-source-id: d73e14a411cdbec815866b02e75ffd71a9186e89
2020-01-10 17:53:41 -08:00
..
api C++ tensor indexing: add Slice / TensorIndex (#30424) 2020-01-10 17:53:41 -08:00
common Trim libshm deps, move tempfile.h to c10 (#17019) 2019-02-13 19:38:35 -08:00
dist_autograd Remove unused argument "destId" in addSendRpcBackward (#31207) 2019-12-14 20:28:29 -08:00
jit check for object equality in constant pooling (#31800) 2020-01-08 16:47:07 -08:00
rpc Avoid sending large unneeded data over wire in process_group_agent. (#31357) 2019-12-18 19:24:24 -08:00
__init__.py Add train() / eval() / is_training() to C++ ScriptModule API (#16044) 2019-02-01 13:07:38 -08:00