Torch7从Tensor访问一个元素作为Tensor

Question

我正在使用Torch7来训练一些神经网络，并且我有一个昏暗的Timor 1（向量），我想在这个向量中访问元素i。 不幸的是，它给了我一个整数而不是大小为1的Tensor。

我明白了：

matrix = torch.Tensor{1,2}
>  1
   2
  [torch.DoubleTensor of size 2]
matrix[1]
> 1

我想要这个：

matrix[1]
> 1 
  [torch.DoubleTensor of size 1]

我有责任这样做：

torch.Tensor{matrix[1]}
> 1 
  [torch.DoubleTensor of size 1]   

Answer 1

您可以使用torch indexing operator，如下所示：

> t = matrix[{ {1} }]
> = t
 1
[torch.DoubleTensor of size 1]