tf.random.experimental.stateless_shuffle

Случайным и детерминированным образом перемешивает тензор вдоль первого измерения.

tf.random.experimental.stateless_shuffle(
    value, seed, alg='auto_select', name=None
)

Тензор перемешивается вдоль измерения 0 так, что каждому value[j] соответствует ровно один output[i]. Например, для тензора размером 3x2 может быть выполнено такое отображение:

[[1, 2],       [[5, 6],
 [3, 4],  ==>   [1, 2],
 [5, 6]]        [3, 4]]

v = tf.constant([[1, 2], [3, 4], [5, 6]])
shuffled = tf.random.experimental.stateless_shuffle(v, seed=[8, 9])
print(shuffled)
tf.Tensor(
[[5 6]
  [1 2]
  [3 4]], shape=(3, 2), dtype=int32)

Это версия tf.random.shuffle без сохранения состояния: при двух запусках с одинаковыми value и seed результат будет одинаковым. Результат не меняется при повторных запусках на одном и том же оборудовании (а также между CPU и GPU), но может измениться между версиями TensorFlow или при работе на оборудовании, отличном от CPU/GPU.