import tensorflow as tf

              kernel_size=[5, 5],
              padding="same",
              activation=tf.nn.relu)
            pool1 = tf.layers.max_pooling2d(inputs=conv1, pool_size=[2, 2], strides=2)
            conv2 = tf.layers.conv2d(
                inputs=pool1,
                filters=filter_list[1],
                kernel_size=[5, 5],
                padding="same",
                activation=tf.nn.relu)
            pool2 = tf.layers.max_pooling2d(inputs=conv2, pool_size=[2, 2], strides=2)
        
#     return int(np.prod(pool2.get_shape().as_list()[1:]))
    return pool2.get_shape().as_list()

if __name__ == "__main__":
    tf.autograph.set_verbosity(0)
    print(get_conv_dimension([32, 64]))