python - 从“tf.experimental.numpy.random.randint”中获取整数值

Question

我想使用 tensorflow 生成一个随机整数。下面是我在 tensorflow nightly version 中找到的代码。如何从此代码中获取整数值。或者有什么方法可以在张量流中生成随机整数

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

标签： pythontensorflowtensor

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”，而无需检索其无延迟值。