1

私はいくつかのニューラル ネットをトレーニングするために Torch7 に取り組んでおり、dim 1 (ベクトル) の Tensor を取得しており、このベクトルの要素 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]   
4

1 に答える 1

1

torch インデックス演算子は次のように使用できます。

> t = matrix[{ {1} }]
> = t
 1
[torch.DoubleTensor of size 1]
于 2015-07-03T15:51:55.597 に答える