1-d IntTensorをPytorchのintにキャストする方法

Question

1次元のIntTensorを取得しましたが、整数に変換したいと思います。私はこの方法でそれを試します：

print(dictionary[IntTensor.int()])

エラーが発生しました：

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

ありがとう〜

skb · Accepted Answer

以下を使用できます。

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

使用しているキーは、タイプautograd.Variableのオブジェクトです。 .dataはテンソルを提供し、インデックス0を使用して要素にアクセスできます。

Marcin · Answer

私が知っている最も簡単でクリーンな方法：

IntTensor.item()

PyTorch docsから：

"このテンソルの値を標準として返しますPython数値。これは、1つの要素を持つテンソルに対してのみ機能します。他の場合については、：meth：~Tensor.tolistを参照してください。"

Virkillthenasty · Answer

torch.tensor('variable',dtype=torch.int8)

これを試して