tensorflow gravity

gravity.py

from datetime import datetime
import tensorflow as tf

now = datetime.utcnow().strftime("%Y%m%d%H%M%S")
root_logdir = "tf_logs"
logdir = "{}/run-{}/".format(root_logdir, now)

n_epochs = 20
time_step = 0.1

initial_positions = tf.random_uniform([1, 5], 20, 100)
position = tf.Variable(initial_positions)
#position = tf.Variable([5,5,5,5,5], dtype=tf.float32, name="position")
velocity = tf.Variable([0,0,0,0,0], dtype=tf.float32, name="velocity")

gravity = tf.constant(9.8, dtype=tf.float32, name="gravity")

init = tf.global_variables_initializer()

velocity_update = tf.assign(velocity, velocity + gravity * time_step)
position_update = tf.assign(position, position + velocity_update * time_step)

position_summary = tf.summary.scalar('position', position)
file_writer = tf.summary.FileWriter(logdir, tf.get_default_graph())

with tf.Session() as sess:
    sess.run(init)

    for epoch in range(n_epochs):
        sess.run(position_update)
        print(position.eval())
        summary_str = position_summary.eval(feed_dict={X: epoch, y: position})