torch.Tensor.index_select

Tensor.index_select(dim, index) → Tensor

参见 torch.index_select()