2.7 序列化(保存)tensor


可以直接調用tensor的save和load方法對tensor進行保存和讀取.

保存:

 

 讀取:

 

 或者

 

 但是這樣保存下來的只能被Pytorch所識別。可以利用 h5py 庫進行通用格式的保存。

首先要轉換成numpy數組,然后調用  create_dataset 函數,保存為hdf5格式的文件

 

 這里的 coords是我們定義的key值。hdf5格式的文件是基於key-value保存的,所以我們在保存的時候要指定一個key。

hdf5文件支持從硬盤上部分讀取,比如我們只讀取coords的部分數據

當文件指針被打開,或者數據集被加載時,pytor不會馬上加載數據,之后當我們需要取具體的數據時,才會從硬盤上讀取我們

所需要的數據。讀取到的數據可以看做是一個numpy數組。因此我們可以直接把讀取到的數據轉換成tensor。

 

 最后,記得關閉文件。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM