Tensorflow：如何使用 tensorflow 操作从给定的字符串列表中选择一个随机字符串

Question

我有一个像下面这样的字符串列表，我想做张量流操作。 对于来自list_1的给定输入字符串，我想从list_2随机选择一个字符串。

我的功能是这样的；

list_1 = ['hello', 'how are you?', 'this is just a test']
list_2 = ['helloooo', 'thanks', 'okay']

def test(string):
    if string in list_1:
        print("list: ", random.choice(list_2)) 
test("hello") 

我如何使用 tensorflow 实现这一目标？ 因为我不觉得任何东西tf.random.choice在tensorflow。

Answer 1

这是一种基于 Tensorflow Ops 的不同方法：

import tensorflow as tf

list_2 = ['helloooo', 'thanks', 'okay']
len_list = tf.size(list_2)
rand_var = tf.random_uniform([1],0,len_list, dtype=tf.int32)
output = tf.gather(list_2,rand_var[0])

with tf.Session() as sess:
    print(sess.run(output))