def _maybe_broadcast_volatility(volatility_parts,
                                state_parts):
  """"""
  return [v + tf.zeros_like(sp, dtype=sp.dtype.base_dtype)
          for v, sp in zip(volatility_parts, state_parts)]