(self, state, sess=None)
| 55 | self.output = tf.squeeze(self.output) |
| 56 | |
| 57 | def transform(self, state, sess=None): |
| 58 | sess = sess or tf.get_default_session() |
| 59 | return sess.run(self.output, { self.input_state: state }) |
| 60 | |
| 61 | |
| 62 | def update_state(state, obs_small): |