tf.random_index_shuffle

Outputs the position of value in a permutation of [0, ..., max_index].

Output values are a bijection of the index for any combination and seed and max_index.

If multiple inputs are vectors (matrix in case of seed) then the size of the first dimension must match.

The outputs are deterministic.

index A Tensor. Must be one of the following types: int32, uint32, int64, uint64. A scalar tensor or a vector of dtype dtype. The index (or indices) to be shuffled. Must be within [0, max_index].
seed A Tensor. Must be one of the following types: int32, uint32, int64, uint64. A tensor of dtype Tseed and shape [3] or [n, 3]. The random seed.
max_index A Tensor. Must have the same type as index. A scalar tensor or vector of dtype dtype. The upper bound(s) of the interval (inclusive).
name A name for the operation (optional).

A Tensor. Has the same type as index.