查阅资料发现需要将None 替换成-1
output = tf.reshape(input,input.get_shape().as_list(), name="flatten")
更改为以下代码
output = tf.reshape(input, [-1, input.get_shape().as_list()[1],input.get_shape().as_list()[2],input.get_shape().as_list()[3]], name="flatten")
OK啦