在torch中，如何用预定义的顺序分割张量？

Question

我有一个10 train = torch.range(1,10)的数据集。我想将它按p = torch.randperm(10)定义的随机顺序分割。

要按范围获得切片，可以执行a = train[{{1,3}}]来获取元素的前三个元素。但让我说，我想要第二，第三和第九元素。我能在不像这样操作for循环的情况下得到这个吗

for i = 1,3 do
  print(a[{ p[i] }])
end

哪里

p[1] = 2, p[2] = 3, p[3] = 9. 

a = train[{{ p[{{1,3}}] }}]不工作。

Answer 1

首先，有索引，但是它需要longTensors：

train = torch.range(1,10)
p = torch.randperm(10):long()
print(train:index(p))