使用指数的火炬提取子张量

Question

我想从原始张量中提取一个subtensor，它包含由另一个张量定义的索引处的元素。

说

Integer

如何使用i？

中的索引查询ls

Answer 1

sub = ls:index(1, torch.LongTensor{2, 7, 9})

请参阅[Tensor] index(dim, index)。