2017-12-01 5 views
0

1次元IntTensorを取得しましたが、整数に変換します。 私はこの方法でそれを試してみてください。Pytorchで1次元IntTensorをintにキャストする方法

print(dictionary[IntTensor.int()]) 

が、エラーました:あなたが使用することができます

KeyError: Variable containing: 
423 
[torch.IntTensor of size 1] 

感謝〜

答えて

0

:キーあなた

print(dictionary[IntTensor.data[0]])

」を使用するのはタイプautograd.Variableのオブジェクトです。 .dataテンソルとインデックスを与える0要素にアクセスするために使用することができます。

+0

ありがとうございます。 – Ruben

関連する問題