我正在尝试在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的新手