Method transform

rl2/a3c/worker.py:35–37

(self, state, sess=None)

      self.output = tf.squeeze(self.output)

  def transform(self, state, sess=None):
    sess = sess or tf.get_default_session()
    return sess.run(self.output, { self.input_state: state })


# Create initial state by repeating the same frame 4 times