Tiếp cận RNN weights- Tensorflow

Question

Tôi đang sử dụng tf.python.ops.rnn_cell.GRUCell

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

Làm thế nào để có được trọng lượng của GRUCell này. Tôi cần xem chúng để gỡ lỗi.

Answer 1

Các giá trị của tất cả các biến trong phiên hiện tại có thể được in sử dụng:

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)