lua+torch 索引

lua+torch 索引

参考官方的文档 https://github.com/torch/torch7/blob/master/doc/tensor.md

-- about 索引
A = torch.Tensor({{10., -1., 2., 0.},
                  {-1., 11., -1., 3.},
                  {2., -1., 10., -1.},
                  {0., 3., -1., 8.}})
b = torch.Tensor({6., 25., -11., 15.})

--print(A[1])
print(A[{ 3,{1,3}}]) --partially slice!!!
print(b[{{1,3}}])