import tensorflow as tf

        h_pool5 = h_conv5

     

        h_pool5 = tf.reshape(h_pool5, shape=[-1, POOL_X, POOL_Y])
        feature_mat = h_pool5
        print("----feature_mat-----")
        print(feature_mat)
