【问题标题】:Torch7 Access one element from a Tensor as a TensorTorch7 从张量中访问一个元素作为张量
【发布时间】:2015-09-21 04:06:28
【问题描述】:

我正在使用 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]   

【问题讨论】:

    标签: torch


    【解决方案1】:

    您可以使用torch indexing operator,如下:

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2020-09-02
      • 2020-09-11
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多