如何在张量流中生成随机混洗数

Question

我想生成从0到特定数字的数字，然后随机地随机播放这些数字。 我从张量total = tf.Variable(BCIsize[0]+BCRsize[0]+BCOsize[0], dtype = tf.int32)得到一些价值。例如，如果total的值为20，我想生成0到19之间的数字。

我尝试使用reqnum= tf.random_uniform([1,total],minval=0,maxval=total,tf.int32)生成 但它Tensor("random_uniform:0", shape=(1, ?), dtype=int32)。

我需要生成相同的数字，然后将其作为“总计”的值进行随机播放。变量。 我期待随机输出像[8,14,1,3,5,11,6,9,10,19,12,13,2,15,7,16,17,4,18]

Answer 1

使用tf.range创建列表，然后使用tf.random_shuffle

idx = tf.constant(20)
idx_list = tf.range(idx) # 0~19
shuffle = tf.random_shuffle(idx_list)

with tf.Session() as sess:
    a, b = sess.run([idx_list, shuffle])