Torch7 从张量中访问一个元素作为张量

Question

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

我知道了：

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]

Torch7 Access one element from a Tensor as a Tensor