IT序号网

python访问 RNN 权重Tensorflow

cloudgamer 2023年05月26日 程序员 17 0

我正在使用 tf.python.ops.rnn_cell.GRUCell

output, state = tf.nn.dynamic_rnn( 
        GRUCell(HID_DIM), 
        sequence, 
        dtype=tf.float32, 
        sequence_length=length(sequence) 
) 

我如何获得这个 GRUCell 的权重。我需要查看它们以进行调试。

请您参考如下方法:

可以使用以下命令打印当前 session 中所有变量的值:

with tf.Session() as sess: 
    variables_names =[v.name for v in tf.trainable_variables()] 
    values = sess.run(variables_names) 
    for k,v in zip(variables_names, values): 
        print(k, v) 


评论关闭
IT序号网

微信公众号号:IT虾米 (左侧二维码扫一扫)欢迎添加!