pytochモデルの予測結果とdarrayの相互転送方式

415 ワード

pytoch モデル予測結果 darray

予測結果はnumpyに変わります。


logits=model(feature)
 
#       GPU 
result=logits.data.cpu().numpy()  /  logits.cpu().numpy()
 
#      cpu 
result=logits.data.numpy()  /  logits.numpy()

行列をtensorに変換:


np_arr = np.array([1,2,3,4])
tensor=torch.from_numpy(np_arr)

以上のpytouchモデルの予測結果とdarrayの転送方法は小編集が皆さんに共有する内容です。参考にしていただければと思います。よろしくお願いします。

pytouch、cuda、pythonのバージョンの配置問題を簡単に話します。

Spring beanロード実行順序例解析