Return random integers from low (inclusive) to high (exclusive). Interface borrowed from numpy.random.randint.

randint.py

def randint(low, high=None, size=None, dtype=tf.int32):
    """
    Return random integers from low (inclusive) to high (exclusive).
    Interface borrowed from numpy.random.randint.
    """
    if high is None:
        high = low
        low = 0
    assert dtype.is_integer(), 'dtype for randint must be an integer'
    return tf.random_uniform(shape=size, minval=low, maxval=high, dtype=dtype)