nvidia.dali.experimental.dynamic.element_extract

nvidia.dali.experimental.dynamic.element_extract(input, /, *, batch_size=None, device=None, element_map)

Extracts one or more elements from input sequence.

The outputs are slices in the first (outermost) dimension of the input. There are as many outputs as the elements provided in the element_map.

For example, for element_map = [2, 0, 3] there will be three outputs, containing 2nd, 0th and 3rd element of the input sequences respectively.

The input layout, if provided, must begin with F dimension. The outputs will have one less dimension than the input, that is for FHWC inputs, the outputs will be HWC elements.

This operator expects sequence inputs.

Supported backends

• ‘cpu’

• ‘gpu’

Parameters:

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

Keyword Arguments:

element_map (int or list of int) – Indices of the elements to extract.