保存

有一個(gè)tensor數(shù)據(jù)需要保存，這時(shí)訓(xùn)練好的一個(gè)節(jié)點(diǎn)embedding：

我們需要把結(jié)果保存下來(lái)：

剛開(kāi)始使用的是：

np.savetxt('./obj/model.csv',model.encode().numpy(),fmt='%.2f',delimiter=',')

但是運(yùn)行報(bào)錯(cuò)：

提示說(shuō)要使用tensor.detach().numpy()

detach()方法的作用是從計(jì)算圖中脫離出來(lái)。返回一個(gè)新的tensor，新的tensor和原來(lái)的tensor共享數(shù)據(jù)內(nèi)存，但不涉及梯度計(jì)算，即requires_grad=False。

因此需要改成np.savetxt('./obj/model.csv',model.encode().detach().numpy(),fmt='%.2f',delimiter=',')就可以了。

• fmt : 寫(xiě)入文件的格式,例如：%d %.2f %.18e
• delimiter :分割字符串,默認(rèn)是任何空格。 例如: a = np.arange(100).reshape(5,20)

加載

np.loadtxt(fname,dtype=np.float,delimiter=None,unpack=False)

• frame : 文件、字符串或產(chǎn)生器,可以是.gz 或.bz2的壓縮文件
• dtype : 數(shù)據(jù)類型,可選
• delimiter :分割字符串,默認(rèn)是任何空格。
• unpack :讀入數(shù)據(jù)寫(xiě)入一個(gè)數(shù)組 如果是True,讀入屬性將分別寫(xiě)入不同變量.

例如： b = np.loadtxt(‘a(chǎn).csv’,delimiter=’,’)

b = np.loadtxt(‘a(chǎn).csv’,dtype = np.int,delimiter=’,’)

總結(jié)

以上是生活随笔為你收集整理的Pytorch教程（十八）tensor的保存为csv，并加载的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。