tf.random.experimental.index_shuffle

Возвращает позицию index в перестановке [0, ..., max_index].

tf.random.experimental.index_shuffle(
    index, seed, max_index
)

Для каждой возможной seed и max_index существует одна псевдослучайная перестановка последовательности S=[0, ..., max_index]. Вместо создания полного массива мы можем вычислить новую позицию любого целого числа i (0 <= i <= max_index) в S. Это может быть полезно для очень больших max_index, избегая выделения больших блоков памяти.

В простейшем случае, index и max_index являются скалярами, а seed — вектором длиной 2 (как обычно для бессостоятельных генераторов случайных чисел). Но вы можете добавить ведущий пакетный размер ко всем им. Если у некоторых из них нет пакетного измерения, а у других есть, index_shuffle добавит пакетное измерение к первым путем трансляции.

Входные данные index и выходные данные могут использоваться в качестве индексов для перестановки вектора. Например:

vector = tf.constant(['e0', 'e1', 'e2', 'e3'])
indices = tf.random.experimental.index_shuffle(
  index=tf.range(4), seed=[5, 9], max_index=3)
print(indices)
tf.Tensor([2 0 1 3], shape=(4,), dtype=int32)
shuffled_vector = tf.gather(vector, indices)
print(shuffled_vector)
tf.Tensor([b'e2' b'e0' b'e1' b'e3'], shape=(4,), dtype=string)

Более полезно, его можно использовать в потоковом (также онлайн) сценарии, например, в tf.data, где каждый элемент vector обрабатывается индивидуально, а весь vector никогда не материализуется в памяти.

dataset = tf.data.Dataset.range(10)
dataset = dataset.map(
 lambda idx: tf.random.experimental.index_shuffle(idx, [5, 8], 9))
print(list(dataset.as_numpy_iterator()))
[3, 8, 0, 1, 2, 7, 6, 9, 4, 5]

Эта операция является бессостоятельной (как и функции tf.random.stateless_*), что означает, что вывод полностью определяется seed (при равных других входных данных). Каждый выбор seed соответствует одной перестановке, поэтому при многократном вызове этой функции для одной и той же перестановки убедитесь, что вы используете тот же seed. Например:

seed = [5, 9]
idx0 = tf.random.experimental.index_shuffle(0, seed, 3)
idx1 = tf.random.experimental.index_shuffle(1, seed, 3)
idx2 = tf.random.experimental.index_shuffle(2, seed, 3)
idx3 = tf.random.experimental.index_shuffle(3, seed, 3)
shuffled_vector = tf.gather(vector, [idx0, idx1, idx2, idx3])
print(shuffled_vector)
tf.Tensor([b'e2' b'e0' b'e1' b'e3'], shape=(4,), dtype=string)