nvidia.dali.experimental.dynamic.batch_permutation#

nvidia.dali.experimental.dynamic.batch_permutation(
*,
batch_size=None,
device=None,
allow_repetitions=None,
no_fixed_points=None,
seed=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.

  • 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.

On this page