python - understanding - Accediendo a pesas RNN- Tensorflow

lstm tensorflow example

Question

Estoy usando tf.python.ops.rnn_cell.GRUCell

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

¿Cómo puedo obtener los pesos de este GRUCell? Necesito verlos para la depuración.

Answer 1

Los valores de todas las variables en la sesión actual pueden imprimirse usando:

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)