张量流中获取random.sample的最佳方法

时间:2018-07-20 10:51:11

标签: python-3.x performance tensorflow

我正在尝试在tensorflow会话中复制此python代码

  random.sample(range(0, total_samples), selected_samples)

我可以轻松地使用tf.range函数来生成范围,但是没有tf.random_sample函数

我想出了一个解决方案,但是我不确定这是否是最佳的性能选择(因为我将所有样本都混洗了)

我的解决方案:

all_samples = tf.range(0, total_samples)
shuffled_samples = tf.random_shuffle(all_samples)
random_sample = shuffled_samples[:selected_samples]

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    random_samples = sess.run(random_sample)

如果这是最佳解决方案,请将其张贴在下面,请接受。否则,请让我知道更好的方法,因为我是tensorflow的新手

0 个答案:

没有答案