Lua 如何在torch中按预定义顺序切片张量？

我有一个长度为10的数据集train=torch.range1,10。我想按照p=torch.randperm10定义的随机顺序对其进行切片

要按范围获取切片，可以执行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=列车[{p[{{1,3}]}]不工作

不过，对于其中一个，它需要长传感器：

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

可能重复的