(state, num)
return states
Creates new independent RNG states from an existing state.

  Args:
    state: the existing state.
    num: the number of the new states.

  Returns:
    A tuple of new states.
Creates new independent RNG states from an existing state.
Creates new independent RNG states from an existing state .
def split(state, num):
  """Creates new independent RNG states from an existing state.

  Args:
    state: the existing state.
    num: the number of the new states.

  Returns:
    A tuple of new states.
  """
  state = tf_np.asarray(state, dtype=_RNG_KEY_DTYPE)
  state = _key2seed(state)
  try:
    states = tf.random.experimental.stateless_split(state, num)
  except AttributeError as e:  # pylint: disable=unused-variable
    # TODO(afrozm): For TF < 2.3 we need to do this. Delete once 2.3 launches.
    states = stateless_split(state, num)
  states = tf.unstack(states, num)
  states = tf.nest.map_structure(_seed2key, states)
  return states
def split ( state , num ) : state = tf_np . asarray ( state , dtype = _RNG_KEY_DTYPE ) state = _key2seed ( state ) try : states = tf . random . experimental . stateless_split ( state , num ) except AttributeError as e : # pylint: disable=unused-variable # TODO(afrozm): For TF < 2.3 we need to do this. Delete once 2.3 launches. states = stateless_split ( state , num ) states = tf . unstack ( states , num ) states = tf . nest . map_structure ( _seed2key , states ) return states