在TensorFlow中生成随机整数

Question

我希望在TensorFlow中生成随机整数，但我不知道应该使用哪个命令。特别地，我希望从取值为{1、2、3、4}的均匀分布随机变量中生成。我尝试在tensorflow_probability的分布中查找，但没有找到。

感谢您的帮助。

- Maurizio Serra

Answer 1

如果您想要从均匀分布中获取简单整数，可以使用tf.random.uniform。

为了获得特定的范围和整数，您应该指定minval、maxval和dtype参数。所以在您的情况下:

对于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))

请注意，实际最大值将是maxval-1。

- sebastian-sz