web-dev-qa-db-ja.com

リストまたは派手な配列を1Dトーチテンソルに変換する方法は?

Float値のリスト(またはnumpy配列)があります。これらすべての値を含む1Dトーチテンソルを作成したいと思います。トーチテンソルを作成し、値を格納するループを実行できます。

しかし、私は何らかの方法があることを知りたいのですが、リストまたは配列から初期値を使用してトーチテンソルを作成できますか?私がpytorchで作業しているときにこれを達成するためのPython的な方法があるかどうかも教えてください。

11
Wasi Ahmad

これらはpytorchの一般的な操作であり、ドキュメントで利用できます。 PyTorchはnumpyとの簡単なインターフェースを可能にします。 from_numpyというメソッドがあり、ドキュメントが利用可能です here

import numpy as np 
import torch 
array = np.arange(1, 11)
tensor = torch.from_numpy(array)
16
Kashyap