tf.random_index_shuffle | TensorFlow v2.16.1 (original) (raw)

tf.random_index_shuffle

Stay organized with collections Save and categorize content based on your preferences.

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

View aliases

Compat aliases for migration

tf.compat.v1.random_index_shuffle

tf.random_index_shuffle(
    index: Annotated[Any, TV_RandomIndexShuffle_dtype],
    seed: Annotated[Any, TV_RandomIndexShuffle_Tseed],
    max_index: Annotated[Any, TV_RandomIndexShuffle_dtype],
    rounds: int = 4,
    name=None
) -> Annotated[Any, TV_RandomIndexShuffle_dtype]

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.

Args
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).
rounds	An optional int. Defaults to 4. The number of rounds to use the in block cipher.
name	A name for the operation (optional).

Returns
A Tensor. Has the same type as index.