时间戳

首页 > 解决方案 > 在张量流中以另一个张量作为种子生成随机序列

python - 在张量流中以另一个张量作为种子生成随机序列

问题描述

我有一个用例，我需要在给定输入整数的情况下生成一系列随机整数。在 python 中有很多方法可以做到这一点。我目前使用的一个如下：

import hashlib
def nextRandom(seed, length, maxval):
    md5 = hashlib.md5(str(hash(seed)).encode('utf-8'))
    for k in range(length):
        md5.update(str(k).encode('utf-8'))
        yield int(md5.hexdigest(), 16) % maxval

seed = 12345
length = 10
maxval = 1000
for randInt in nextRandom(seed, length, maxval):
    print(randInt)

这确保生成的序列在给定seed值的情况下是固定的。

现在，我需要在 tensorflow 中使用类似的功能，其中seed作为张量，返回的序列也应该是张量。

我在 tensorflow github 页面中检查了这个问题，但找不到有效的解决方案。

标签： pythontensorflowrandomrandom-seedhashlib

解决方案

这是一种可能的tf.contrib.stateless模块解决方案：

import tensorflow as tf

seed = tf.random_uniform(shape=[2], dtype=tf.int64, maxval=1000000)
x = tf.contrib.stateless.stateless_random_uniform(shape=(1, 1), dtype=tf.float32, seed=seed)

sess = tf.Session()
print(sess.run(x))

请注意，stateless模块不支持整数，除非stateless_multinomial以防万一。

推荐阅读