Method add_noise

acme/tf/networks/noise.py:35–39

(tensor: tf.Tensor)

  def __call__(self, inputs: types.NestedTensor) -> types.NestedTensor:
    def add_noise(tensor: tf.Tensor):
      output = tensor + tf.cast(self._noise.sample(tensor.shape),
                                dtype=tensor.dtype)
      output = tf.clip_by_value(output, -1.0, 1.0)
      return output

    return tree.map_structure(add_noise, inputs)