if observation_and_action_constraint_splitter is not None:
observation, mask = observation_and_action_constraint_splitter(
observation)
observation = tf.cast(observation, dtype=self._dtype)
if self._add_bias:
// The bias is added via a constant 1 feature.
observation = tf.concat([
observation,
After Change
if observation_and_action_constraint_splitter is not None:
observation, mask = observation_and_action_constraint_splitter(
observation)
observation = tf.nest.map_structure(lambda o: tf.cast(o, dtype=self._dtype),
observation)
global_observation, arm_observations = self._split_observation(observation)
if self._add_bias:
// The bias is added via a constant 1 feature.
