从"tf.experimental.numpy.random.randint“获取整数值

Question

我想使用tensorflow生成一个随机整数。下面是我在tensorflow夜间版本中找到的代码。如何从这段代码中获取整数值。或者，有没有办法在tensorflow中生成随机整数

tf.experimental.numpy.random.randint(low=0, high=5, size=1,dtype=np.int)

Answer 1

a = tf.experimental.numpy.random.randint(low=0, high=5, size=1,dtype=np.int)

a.tolist()
>>> [2]

int(a)
>>> 2

np.array(a)
>>> array([2])

np.array(a)[0]
>>> 2

对于TF 2.3.0：

g1 = tf.random.Generator.from_seed(1)
a = g1.uniform([1], minval=0, maxval=15, dtype=tf.dtypes.int32, name=None)

print(a)
>>> tf.Tensor([8], shape=(1,), dtype=int32)

a.numpy()
>>>array([8], dtype=int32)

此外，您可以在TF操作中直接使用张量'a‘，而无需检索其未延迟的值。