(a)
return tf_np.asarray(int32s_to_int64(a))
Converts an RNG seed to an RNG key.

  Args:
    a: an RNG seed, a tensor of shape [2] and dtype `tf.int32`.

  Returns:
    an RNG key, an ndarray of shape [] and dtype `np.int64`.
Converts an RNG seed to an RNG key.
Converts an RNG seed to an RNG key .
def _seed2key(a):
  """Converts an RNG seed to an RNG key.

  Args:
    a: an RNG seed, a tensor of shape [2] and dtype `tf.int32`.

  Returns:
    an RNG key, an ndarray of shape [] and dtype `np.int64`.
  """

  def int32s_to_int64(a):
    """Converts an int32 tensor of shape [2] to an int64 tensor of shape []."""
    a = tf.bitwise.bitwise_or(
        tf.cast(a[0], tf.uint64),
        tf.bitwise.left_shift(
            tf.cast(a[1], tf.uint64), tf.constant(32, tf.uint64)))
    a = tf.cast(a, tf.int64)
    return a

  return tf_np.asarray(int32s_to_int64(a))
def _seed2key ( a ) : def int32s_to_int64 ( a ) : """Converts an int32 tensor of shape [2] to an int64 tensor of shape [].""" a = tf . bitwise . bitwise_or ( tf . cast ( a [ 0 ] , tf . uint64 ) , tf . bitwise . left_shift ( tf . cast ( a [ 1 ] , tf . uint64 ) , tf . constant ( 32 , tf . uint64 ) ) ) a = tf . cast ( a , tf . int64 ) return a return tf_np . asarray ( int32s_to_int64 ( a ) )