def testUniform(self): minval = 0.43 maxval = 3.10 shape = [13, 34, 29] atol = 0.1 outputs = extensions.uniform(123, shape, minval=minval, maxval=maxval) self.assertAllClose((minval + maxval) / 2.0, np.mean(outputs), atol=atol)
def tf_randint(key, shape, minval, maxval, dtype=onp.int32): """Sample uniform random values in [minval, maxval) with given shape/dtype. Args: key: a PRNGKey used as the random key. shape: a tuple of nonnegative integers representing the shape. minval: int or array of ints broadcast-compatible with ``shape``, a minimum (inclusive) value for the range. maxval: int or array of ints broadcast-compatible with ``shape``, a maximum (exclusive) value for the range. dtype: optional, an int dtype for the returned values (default int32). Returns: A random array with the specified shape and dtype. """ return tf_np_extensions.uniform(key, shape, minval=minval, maxval=maxval, dtype=dtype)