火炬，如何选择张量的子集？

Question

例如，我有一个张量t = torch.rand(10,1)。我想在{1,3,7,9}的行中选择张量子集并将它们连接为新的张量。有什么优雅的方式吗？谢谢。

Answer 1

有:index方法：

tens = torch.rand(10,1)
ind = torch.LongTensor{1,3,7,9}
subset = tens:index(1, ind)