Simple example of gradient descent in tensorflow

gd_simple.py

import tensorflow as tf

x = tf.Variable(2, name='x', dtype=tf.float32)
log_x = tf.log(x)
log_x_squared = tf.square(log_x)

optimizer = tf.train.GradientDescentOptimizer(0.5)
train = optimizer.minimize(log_x_squared)

init = tf.initialize_all_variables()

def optimize():
  with tf.Session() as session:
    session.run(init)
    print("starting at", "x:", session.run(x), "log(x)^2:", session.run(log_x_squared))
    for step in range(10):  
      session.run(train)
      print("step", step, "x:", session.run(x), "log(x)^2:", session.run(log_x_squared))
        

optimize()

great example..how do you get the optimized variables/parameters from this? ie not the loss but the parameters that result in the lowest loss?

Great example! Thanks!

Thank you! Simple and straightforward

Thank you

Anyone have this for tensorflow 2.0?