tf.random.stateless_categorical

Генерирует детерминированные псевдослучайные выборки из категориального распределения.

tf.random.stateless_categorical(
    logits,
    num_samples,
    seed,
    dtype=tf.dtypes.int64,
    name=None
)

Это бессостоятельная версия tf.categorical: если её запустить дважды с одинаковыми начальными значениями и формами, она сгенерирует те же псевдослучайные числа. Выходные данные согласованы при нескольких запусках на одном оборудовании (и между CPU и GPU), но могут отличаться между версиями TensorFlow или на оборудовании, не являющимся CPU/GPU.

Пример:

# samples has shape [1, 5], where each value is either 0 or 1 with equal
# probability.
samples = tf.random.stateless_categorical(
    tf.math.log([[0.5, 0.5]]), 5, seed=[7, 17])