Генерация случайных целых чисел в TensorFlow

Question

Я хотел бы генерировать случайные целые числа в TensorFlow, но я не знаю, какую команду мне следует использовать. В частности, я хотел бы генерировать из единой случайной переменной, которая принимает значения в {1, 2, 3, 4}. Я попытался найти среди дистрибутивов, включенных в tensorflow_probability, но не нашел его.

Заранее спасибо за вашу помощь.

Answer 1

Для простых целых чисел из равномерного распределения вы можете использовать tf.random.uniform .
В порядке, чтобы получить указанный диапазон и целые числа, вы должны указать minval, maxval и * 1006. * параметры. Так что в вашем случае:

Для Tensorflow 2.0 и выше:

print(tf.random.uniform(shape=(), minval=1, maxval=5, dtype=tf.int32)

Для Tensorflow 1.15 и ниже:

with tf.Session() as sess:
    random_int = tf.random.uniform(shape=(), minval=1, maxval=5, dtype=tf.int32)
    print(sess.run(random_int))