Tensor 索引 API

PyTorch C++ API 中的張量索引與 Python API 非常相似。所有索引型別，如 None / ... / 整數 / 布林值 / 切片 / 張量，在 C++ API 中都可用，這使得從 Python 索引程式碼到 C++ 的轉換非常簡單。主要區別在於，C++ API 中不使用類似於 Python API 語法的 []-運算子，而是使用索引方法：

• torch::Tensor::index （連結）

• torch::Tensor::index_put_ （連結）

另外需要注意的是，None / Ellipsis / Slice 等索引型別位於 torch::indexing 名稱空間中，建議在任何索引程式碼之前新增 using namespace torch::indexing，以便方便地使用這些索引型別。

以下是將 Python 索引程式碼轉換為 C++ 的示例：

Getter

Python	C++（假設 using namespace torch::indexing）
tensor[None]	tensor.index({None})
tensor[Ellipsis, ...]	tensor.index({Ellipsis, "..."})
tensor[1, 2]	tensor.index({1, 2})
tensor[True, False]	tensor.index({true, false})
tensor[1::2]	tensor.index({Slice(1, None, 2)})
tensor[torch.tensor([1, 2])]	tensor.index({torch::tensor({1, 2})})
tensor[..., 0, True, 1::2, torch.tensor([1, 2])]	tensor.index({"...", 0, true, Slice(1, None, 2), torch::tensor({1, 2})})

Setter

Python	C++（假設 using namespace torch::indexing）
tensor[None] = 1	tensor.index_put_({None}, 1)
tensor[Ellipsis, ...] = 1	tensor.index_put_({Ellipsis, "..."}, 1)
tensor[1, 2] = 1	tensor.index_put_({1, 2}, 1)
tensor[True, False] = 1	tensor.index_put_({true, false}, 1)
tensor[1::2] = 1	tensor.index_put_({Slice(1, None, 2)}, 1)
tensor[torch.tensor([1, 2])] = 1	tensor.index_put_({torch::tensor({1, 2})}, 1)
tensor[..., 0, True, 1::2, torch.tensor([1, 2])] = 1	tensor.index_put_({"...", 0, true, Slice(1, None, 2), torch::tensor({1, 2})}, 1)

Python/C++ 索引型別之間的轉換

Python 和 C++ 索引型別之間的一一對應轉換如下：

Python	C++（假設 using namespace torch::indexing）
無	無
Ellipsis	Ellipsis
...	"..."
123	123
真	真
假	假
: 或 ::	Slice() 或 Slice(None, None) 或 Slice(None, None, None)
1: 或 1::	Slice(1, None) 或 Slice(1, None, None)
:3 或 :3:	Slice(None, 3) 或 Slice(None, 3, None)
::2	Slice(None, None, 2)
1:3	Slice(1, 3)
1::2	Slice(1, None, 2)
:3:2	Slice(None, 3, 2)
1:3:2	Slice(1, 3, 2)
torch.tensor([1, 2])	torch::tensor({1, 2})