如何只保存张量而不是 Tensorflow2 中的模型

Question

我已经学习了几个月的 Tensorflow2，但遇到了一些困难。例如，我创建了一个张量，如下所示：import tensorflow as tf v=tf.random.normal((20,30,40))现在我只想将张量 v 保存到合适的文件中。实际上，v 是从 .nc 数据创建的。我使用包“netCDF4”读取它并选择一些变量，其尺寸为（时间，经度，纬度），以连接将它们放入维度为 (time,lon,lat,var_num) 的 v 中。但是v的大小很大（例如，(1000,224,224,5)）。所以我需要保存v，以防多次读取netcdf。我搜索了一些问题，但对我帮助不大，因为它们要么是关于在 tf1.X 中保存变量，要么是关于在 tf 2 中保存模型（或模型中的变量）。所以我来这里寻求英俊的人的帮助。非常感谢。

Answer 1

您仍然可以使用保存的模型格式来存储单个tf.Variable

您只需捕获vatf.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')