def cast_int(x): """ Cast the output of x to an int64 """ return tf.to_int64(x) # Q learning DQN style