我的主要方法调用模型代码即mnist_small(如下所述)来预测输出,出于某种原因,我想获得“最终权重矩阵”。
我尝试使用
with tf.variable_scope('dense_out', reuse=True):
w = tf.convert_to_tensor('weights')
但没有成功
def mnist_small(inputs):
hidden = tf.layers.dense(name="dense_hidden",
inputs=inputs,
units=500,
activation=tf.nn.leaky_relu,
use_bias=True,
kernel_initializer=tf.initializers.glorot_normal,
bias_initializer=tf.initializers.zeros)
outputs = tf.layers.dense(name="dense_out",
inputs=hidden,
units=10,
activation=tf.nn.leaky_relu,
use_bias=True,
kernel_initializer=tf.initializers.glorot_normal,
bias_initializer=tf.initializers.zeros,
)
return outputs
我希望矩阵具有最终权重