python - Tensorflow中的numpy随机选择

Question

Tensorflow 中是否有与 numpy 随机选择等效的函数。在 numpy 中，我们可以从给定列表中随机获取一个项目及其权重。

 np.random.choice([1,2,3,5], 1, p=[0.1, 0, 0.3, 0.6, 0])

此代码将从给定列表中选择具有 p 个权重的项目。

Answer 1

不，但您可以使用 tf.multinomial 获得相同的结果:

elems = tf.convert_to_tensor([1,2,3,5])
samples = tf.multinomial(tf.log([[1, 0, 0.3, 0.6]]), 1) # note log-prob
elems[tf.cast(samples[0][0], tf.int32)].eval()
Out: 1
elems[tf.cast(samples[0][0], tf.int32)].eval()
Out: 5

[0][0] 部分在这里，因为 multinomial 期望批处理的每个元素都有一行未标准化的对数概率，并且还有另一个维度样本数。