2017-03-20 3 views
2

私は浮動小数点値のリスト(または、数が少ない配列)を持っています。私はすべての値を含む1dトーチテンソルを作成したいと思います。私はトーチテンソルを作成し、ループを実行して値を格納することができます。リストまたはnumpy配列を1dトーチテンソルに変換するには?

しかし、私はそこに何らかの方法があることを知りたいと思います。リストや配列から初期値でトーチテンソルを作成できますか?また、私がpytorchで働いているときに、これを達成するためのpythonicな方法があれば私に提案してください。

答えて

8

これらは一般的なオペレーションであり、ドキュメントに記載されています。 PyTorchはnumpyとの簡単なインターフェースを可能にします。 from_numpyというメソッドがあり、そのドキュメントは利用可能です。here

import numpy as np 
import torch 
array = np.arange(1, 11) 
tensor = torch.from_numpy(array) 
関連する問題