tensorflow BasicRNNCell调试

时间：2023-03-09 06:08:30
tensorflow BasicRNNCell调试

运行以下代码，进入~/anaconda3/lib/python3.5/site-packages/tensorflow/python/ops/rnn.py和~/anaconda3/lib/python3.5/site-packages/tensorflow/python/ops/rnn_cell.py单步追踪调试

调试中import tensorflow as tf,利用tf.Session().run(variable)打印变量

查看BasicRNNCell和dynamic_rnn的实现方式

 #-*-coding:utf8-*-

 __author = "buyizhiyou"

 __date = "2017-11-20"

 '''

 单步调试,学习rnn的tf实现

 '''

 import tensorflow as tf

 import numpy as np

 import pdb  

 X = tf.random_normal(shape=[2,3,4], dtype=tf.float32)#(2,3,4)==>(Batch_size,Time_steps(序列长度),Data_Vector)

 pdb.set_trace()

 cell = tf.nn.rnn_cell.BasicRNNCell(10)#output_size:10,也可以换成GRUCell，LSTMAACell,BasicRNNCell

 state = cell.zero_state(2, tf.float32)#batch_size:2

 output, state = tf.nn.dynamic_rnn(cell, X, initial_state=state, time_major=False)

 with tf.Session() as sess:

     sess.run(tf.global_variables_initializer())

     print (output.get_shape())

     print (sess.run(state))

相关文章