Accès RNN weights- tensorflow

Question

J'utilise tf.python.ops.rnn_cell.GRUCell

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

Comment puis-je obtenir les poids de cette GRUCell. J'ai besoin de les voir pour le débogage.

Answer 1

Les valeurs de toutes les variables de la session en cours peuvent être imprimés à l'aide:

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)