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

Question

我正在使用 Torch7 来训练一些神经网络，我有一个昏暗 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

您可以按如下方式使用火炬索引运算符：

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