Question

我使用 tensorflow 构建了神经网络。

在训练它并获得最低成本后，我需要将权重和偏差保存为矩阵，以便在例如 Matlab 中使用。我该怎么做？

当我在完成训练后执行以下代码以保存权重时：

from scipy.io import savemat
savemat("end_Weight1.mat", weights )

我可以保存它，但我得到空矩阵，这意味着矩阵只包含四个变量但没有任何值。我在矩阵中没有得到任何值。

我认为在保存权重或类似的东西之前需要更新权重？

Answer 1

你可以使用

all_variables=tf.global_variables()

在每个批次上从创建的图表中获取您的变量，然后保存它们。

<块引用>

注意：此方法返回所有变量n，因此您可以过滤列表。例如，您可以根据需要通过偏差或权重对其进行过滤。