def step(self, action, indices=None, name=None): if indices is None: indices = np.arange(self.batch_size) with tf.variable_scope(name, default_name='AtariStep'): rew, done = gym_tensorflow_module.environment_step( self.instances, indices, action) return rew, done
def step(self, action, indices=None, name=None): """環境を1step処理する (batch数だけ複数の環境を一気に処理する) :param action: :param indices: :param name: :return: """ if indices is None: indices = np.arange(self.batch_size) with tf.variable_scope(name, default_name='AtariStep'): rew, done = gym_tensorflow_module.environment_step( self.instances, indices, action) return rew, done
def step(self, action, indices=None, name=None): with tf.variable_scope(name, default_name='MazeStep'): #action = tf.Print(action, [action], 'action=') return gym_tensorflow_module.environment_step(self.instances, indices=indices, action=action)