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.