、2D 张量和 3D 张量进行索引。...3D张量
>>> c = torch.arange(0, 9).view([1, 3, 3])
>>> print(c)
tensor([[[0, 1, 2],
[3, 4, 5],...[6, 7, 8]]])
>>> # 获取3D张量的第1个维度且索引号为0的张量子集
>>> print(torch.index_select(c, dim = 0, index...,因此可以通过 PyTorch 中的高级索引来实现。」...([0, 1])) b[:, [0, 1]];
创建 3D 张量 c 的第 1 个维度且索引号为 0 的张量子集: torch.index_select(c, dim = 0, index = torch.tensor