import tensorflow as tf


"""
Test RNN graph single layer
"""
def test_rnn_kstep(test_data_x,test_data_y, preds, rnn_outputs, g, checkpoint, input_prob, output_prob, state_prob, num_test, kstep = 3):
    with tf.Session() as sess:
        sess.run(tf.global_variables_initializer())
        result= {}

        "read the trained graph"
