在张量流中找到NN模型的最终权重

时间:2019-07-22 09:45:25

标签: python-2.7 tensorflow neural-network

我的主要方法调用模型代码即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

我希望矩阵具有最终权重

0 个答案:

没有答案
相关问题