nvidia.dali.experimental.dynamic.batch_permutation

nvidia.dali.experimental.dynamic.batch_permutation(

*,

batch_size=None,

device=None,

allow_repetitions=None,

no_fixed_points=None,

rng=None,

)

Produces a batch of random integers which can be used as indices for indexing samples in the batch.

Supported backends

• ‘cpu’

Keyword Arguments:

• allow_repetitions (bool, optional, default = False) – If true, the output can contain repetitions and omissions.

• no_fixed_points (bool, optional, default = False) – If true, the the output permutation cannot contain fixed points, that is out[i] != i. This argument is ignored when batch size is 1.

• batch_size (int, optional) – The batch size to broadcast input tensors to. Ignored for batch inputs.

• device (device-like, optional) – The device to use for the operation. Must not conflict with the device of the inputs.

• rng (RNG, optional) – A random number generator instance. Can be obtained by calling nvidia.dali.experimental.dynamic.random.RNG(seed)(). If not provided, the default RNG is used.