1. ホーム
  2. python

[解決済み] PyTorchでテンソルの値を取得するにはどうしたらいいですか?

2022-02-14 08:09:24

質問

テンソルを印刷すると、以下のようになります。

>>> x = torch.tensor([3])
>>> print(x)
tensor([3])

同様に、そのインデックスを作成する .data が得られます。

>>> x.data[0]
tensor(3)

値だけを取得するには 3 ?

解決方法は?

テンソルをnumpyに変換する。

x.numpy()[0]