在 TensorFlow 中生成随机整数

Question

我想在 TensorFlow 中生成随机整数，但我不知道应该使用哪个命令。特别是，我想从一个统一的随机变量中生成，该变量的值为 {1, 2, 3, 4}。我试图查看其中包含的发行版，tensorflow_probability但没有找到。

在此先感谢您的帮助。

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。