learning_rate = tf.placeholder(tf.float32, shape=[])
train_step = tf.train.GradientDescentOptimizer(learning_rate=learning_rate
    ).minimize(mse)
How to set adaptive learning rate for GradientDescentOptimizer?