804cf56a5f0e70d39f6447cd2e0be52fbfec6a8c,tf_agents/utils/tensor_normalizer.py,StreamingTensorNormalizer,variables,#StreamingTensorNormalizer#,251

Before Change

  @property
  def variables(self):
    Returns a tuple of tf variables owned by this normalizer.
    return self._count, self._mean_sum, self._var_sum

  def _update_ops(self, tensor, outer_dims):
    Returns a list of ops which update normalizer variables for tensor.

After Change

    Returns a tuple of tf variables owned by this normalizer.
    return (tf.nest.pack_sequence_as(self._tensor_spec, self._count),
            tf.nest.pack_sequence_as(self._tensor_spec, self._mean_sum),
            tf.nest.pack_sequence_as(self._tensor_spec, self._var_sum))

  def _update_ops(self, tensor, outer_dims):
    Returns a list of ops which update normalizer variables for tensor.

Instances