Randomly select a value from a Python list in TensorFlow 2.

tf2_random_index.py

import tensorflow as tf


my_list = ['item1', 'item2', 'item3']

index = tf.random.uniform(
    [],
    maxval=len(my_list),
    dtype=tf.int32
)

index = tf.random.fixed_unigram_candidate_sampler(
    tf.constant(  # true_classes
        [range(len(my_list))],
        dtype=tf.int64
    ),
    len(my_list),  # num_true
    1,  # num_sampled
    False,  # unique
    len(my_list),  # range_max
    unigrams=[1. / len(my_list)] * len(my_list)
)[0]

index = tf.random.uniform_candidate_sampler(
    [range(0, len(my_list))],  # true_classes
    len(my_list),  # num_true
    1,  # num_sampled,
    False,  # unique
    len(my_list)  # range_max
)[0]

# eager execution:
random_value = my_list[index.numpy().item()]

# autograph:
random_value = my_list[index]