tf.raw_ops.StatelessShuffle

Randomly and deterministically shuffles a tensor along its first dimension.

View aliases

Compat aliases for migration

tf.compat.v1.raw_ops.StatelessShuffle

tf.raw_ops.StatelessShuffle(
    value, key, counter, alg, name=None
)

The tensor is shuffled along dimension 0, such that each value[j] is mapped to one and only one output[i]. For example, a mapping that might occur for a 3x2 tensor is:

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

The outputs are a deterministic function of value, key, counter and alg.

Args
`value`	A `Tensor`. The tensor to be shuffled.
`key`	A `Tensor` of type `uint64`. Key for the counter-based RNG algorithm (shape uint64[1]).
`counter`	A `Tensor` of type `uint64`. Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used.
`alg`	A `Tensor` of type `int32`. The RNG algorithm (shape int32[]).
`name`	A name for the operation (optional).

Returns
A `Tensor`. Has the same type as `value`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/StatelessShuffle