1. 程式人生 > 程式設計 >Pytorch Tensor 輸出為txt和mat格式方式

Pytorch Tensor 輸出為txt和mat格式方式

假設result1為tensor格式,首先將其化為array格式(注意只變成numpy還不行),之後存為txt和mat格式

import scipy.io as io
result1 = np.array(result1)
np.savetxt('npresult1.txt',result1)
io.savemat('save.mat',{'result1':result1})

以上這篇Pytorch Tensor 輸出為txt和mat格式方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。