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)