可以直接調用tensor的save和load方法對tensor進行保存和讀取.
保存:
讀取:
或者
但是這樣保存下來的只能被Pytorch所識別。可以利用 h5py 庫進行通用格式的保存。
首先要轉換成numpy數組,然后調用 create_dataset 函數,保存為hdf5格式的文件
這里的 coords是我們定義的key值。hdf5格式的文件是基於key-value保存的,所以我們在保存的時候要指定一個key。
hdf5文件支持從硬盤上部分讀取,比如我們只讀取coords的部分數據
當文件指針被打開,或者數據集被加載時,pytor不會馬上加載數據,之后當我們需要取具體的數據時,才會從硬盤上讀取我們
所需要的數據。讀取到的數據可以看做是一個numpy數組。因此我們可以直接把讀取到的數據轉換成tensor。
最后,記得關閉文件。