溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章主要介紹了使用pytorch怎么對tensor與numpy數組進行轉換，此處通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考價值，需要的朋友可以參考下：
a = torch.ones(2,2) b = a.numpy() c=np.array(a) #也可以轉numpy數組 print(type(a)) print(type(b)) print(a) print(b)
輸出為：
<class ‘torch.Tensor'> <class ‘numpy.ndarray'> tensor([[1., 1.], [1., 1.]]) [[1. 1.] [1. 1.]]
numpy轉tensor：
import torch import numpy as np a = np.ones(5) b = torch.from_numpy(a) c=torch.Tensor(a) #也可以轉pytorch Tensor print(type(a)) print(type(b)) print(a) print(b)
輸出為：
<class ‘numpy.ndarray'> <class ‘torch.Tensor'> [1. 1. 1. 1. 1.] tensor([1., 1., 1., 1., 1.], dtype=torch.float64)
可見pytorch的tensor對象與numpy數組是可以相互轉換的，且numpy數組的默認類型是double
到此這篇關于使用pytorch怎么對tensor與numpy數組進行轉換的文章就介紹到這了,更多相關使用pytorch怎么對tensor與numpy數組進行轉換的內容請搜索億速云以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持億速云！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
