nvidia.dali.experimental.dynamic.subscript_dim_check

nvidia.dali.experimental.dynamic.subscript_dim_check(input, /, *, batch_size=None, device=None, num_subscripts)

Checks that the input has at least num_subscripts dimensions.

This operator is used internally when all indices are empty (:) and just verifies that the input has sufficient number of dimensions and passes through the input.

Supported backends

• ‘cpu’

• ‘gpu’

Parameters:

input (Tensor/Batch) – Input to the operator.

Keyword Arguments:

num_subscripts (int) – Number of subscripts supplied, which is the minimum required in the input.