我已經(jīng)學(xué)習(xí)了幾個月的 Tensorflow2,但遇到了一些困難。例如,我創(chuàng)建了一個張量,如下所示:import tensorflow as tf
v=tf.random.normal((20,30,40))現(xiàn)在我只想將張量 v 保存到合適的文件中。實際上,v 是從 .nc 數(shù)據(jù)創(chuàng)建的。我使用包“netCDF4”讀取它并選擇一些變量,其尺寸為(時間,經(jīng)度,緯度),以連接將它們放入維度為 (time,lon,lat,var_num) 的 v 中。但是v的大小很大(例如,(1000,224,224,5))。所以我需要保存v,以防多次讀取netcdf。我搜索了一些問題,但對我?guī)椭淮?,因為它們要么是關(guān)于在 tf1.X 中保存變量,要么是關(guān)于在 tf 2 中保存模型(或模型中的變量)。所以我來這里尋求英俊的人的幫助。非常感謝。
1 回答

FFIVE
TA貢獻1797條經(jīng)驗 獲得超6個贊
您仍然可以使用保存的模型格式來存儲單個tf.Variable
您只需捕獲v
atf.Variable
然后將其傳遞給tf.saved_model.save
.
也許是這樣的:
v=tf.Variable(tf.random.normal((20,30,40))) tf.saved_model.save(v,?'/path/to/my_var')
然后從保存的版本再次加載:
v_from_file?=?tf.saved_model.load('/path/to/my_var')
添加回答
舉報
0/150
提交
取消