nvidia.dali.experimental.dynamic.tensor_subscript#

nvidia.dali.experimental.dynamic.tensor_subscript(input, /, *, batch_size=None, device=None, at_0=None, at_1=None, at_10=None, at_11=None, at_12=None, at_13=None, at_14=None, at_15=None, at_16=None, at_17=None, at_18=None, at_19=None, at_2=None, at_20=None, at_21=None, at_22=None, at_23=None, at_24=None, at_25=None, at_26=None, at_27=None, at_28=None, at_29=None, at_3=None, at_30=None, at_31=None, at_4=None, at_5=None, at_6=None, at_7=None, at_8=None, at_9=None, hi_0=None, hi_1=None, hi_10=None, hi_11=None, hi_12=None, hi_13=None, hi_14=None, hi_15=None, hi_16=None, hi_17=None, hi_18=None, hi_19=None, hi_2=None, hi_20=None, hi_21=None, hi_22=None, hi_23=None, hi_24=None, hi_25=None, hi_26=None, hi_27=None, hi_28=None, hi_29=None, hi_3=None, hi_30=None, hi_31=None, hi_4=None, hi_5=None, hi_6=None, hi_7=None, hi_8=None, hi_9=None, lo_0=None, lo_1=None, lo_10=None, lo_11=None, lo_12=None, lo_13=None, lo_14=None, lo_15=None, lo_16=None, lo_17=None, lo_18=None, lo_19=None, lo_2=None, lo_20=None, lo_21=None, lo_22=None, lo_23=None, lo_24=None, lo_25=None, lo_26=None, lo_27=None, lo_28=None, lo_29=None, lo_3=None, lo_30=None, lo_31=None, lo_4=None, lo_5=None, lo_6=None, lo_7=None, lo_8=None, lo_9=None, num_subscripts=None, step_0=None, step_1=None, step_10=None, step_11=None, step_12=None, step_13=None, step_14=None, step_15=None, step_16=None, step_17=None, step_18=None, step_19=None, step_2=None, step_20=None, step_21=None, step_22=None, step_23=None, step_24=None, step_25=None, step_26=None, step_27=None, step_28=None, step_29=None, step_3=None, step_30=None, step_31=None, step_4=None, step_5=None, step_6=None, step_7=None, step_8=None, step_9=None)#

Applies NumPy-like indexing to a batch of tensors.

Supported backends

  • ‘cpu’

  • ‘gpu’

Parameters:

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

Keyword Arguments:

  • at_0 (int or Tensor/Batch of int, optional) – Position index

  • at_1 (int or Tensor/Batch of int, optional) – Position index

  • at_10 (int or Tensor/Batch of int, optional) – Position index

  • at_11 (int or Tensor/Batch of int, optional) – Position index

  • at_12 (int or Tensor/Batch of int, optional) – Position index

  • at_13 (int or Tensor/Batch of int, optional) – Position index

  • at_14 (int or Tensor/Batch of int, optional) – Position index

  • at_15 (int or Tensor/Batch of int, optional) – Position index

  • at_16 (int or Tensor/Batch of int, optional) – Position index

  • at_17 (int or Tensor/Batch of int, optional) – Position index

  • at_18 (int or Tensor/Batch of int, optional) – Position index

  • at_19 (int or Tensor/Batch of int, optional) – Position index

  • at_2 (int or Tensor/Batch of int, optional) – Position index

  • at_20 (int or Tensor/Batch of int, optional) – Position index

  • at_21 (int or Tensor/Batch of int, optional) – Position index

  • at_22 (int or Tensor/Batch of int, optional) – Position index

  • at_23 (int or Tensor/Batch of int, optional) – Position index

  • at_24 (int or Tensor/Batch of int, optional) – Position index

  • at_25 (int or Tensor/Batch of int, optional) – Position index

  • at_26 (int or Tensor/Batch of int, optional) – Position index

  • at_27 (int or Tensor/Batch of int, optional) – Position index

  • at_28 (int or Tensor/Batch of int, optional) – Position index

  • at_29 (int or Tensor/Batch of int, optional) – Position index

  • at_3 (int or Tensor/Batch of int, optional) – Position index

  • at_30 (int or Tensor/Batch of int, optional) – Position index

  • at_31 (int or Tensor/Batch of int, optional) – Position index

  • at_4 (int or Tensor/Batch of int, optional) – Position index

  • at_5 (int or Tensor/Batch of int, optional) – Position index

  • at_6 (int or Tensor/Batch of int, optional) – Position index

  • at_7 (int or Tensor/Batch of int, optional) – Position index

  • at_8 (int or Tensor/Batch of int, optional) – Position index

  • at_9 (int or Tensor/Batch of int, optional) – Position index

  • hi_0 (int or Tensor/Batch of int, optional) – Range end

  • hi_1 (int or Tensor/Batch of int, optional) – Range end

  • hi_10 (int or Tensor/Batch of int, optional) – Range end

  • hi_11 (int or Tensor/Batch of int, optional) – Range end

  • hi_12 (int or Tensor/Batch of int, optional) – Range end

  • hi_13 (int or Tensor/Batch of int, optional) – Range end

  • hi_14 (int or Tensor/Batch of int, optional) – Range end

  • hi_15 (int or Tensor/Batch of int, optional) – Range end

  • hi_16 (int or Tensor/Batch of int, optional) – Range end

  • hi_17 (int or Tensor/Batch of int, optional) – Range end

  • hi_18 (int or Tensor/Batch of int, optional) – Range end

  • hi_19 (int or Tensor/Batch of int, optional) – Range end

  • hi_2 (int or Tensor/Batch of int, optional) – Range end

  • hi_20 (int or Tensor/Batch of int, optional) – Range end

  • hi_21 (int or Tensor/Batch of int, optional) – Range end

  • hi_22 (int or Tensor/Batch of int, optional) – Range end

  • hi_23 (int or Tensor/Batch of int, optional) – Range end

  • hi_24 (int or Tensor/Batch of int, optional) – Range end

  • hi_25 (int or Tensor/Batch of int, optional) – Range end

  • hi_26 (int or Tensor/Batch of int, optional) – Range end

  • hi_27 (int or Tensor/Batch of int, optional) – Range end

  • hi_28 (int or Tensor/Batch of int, optional) – Range end

  • hi_29 (int or Tensor/Batch of int, optional) – Range end

  • hi_3 (int or Tensor/Batch of int, optional) – Range end

  • hi_30 (int or Tensor/Batch of int, optional) – Range end

  • hi_31 (int or Tensor/Batch of int, optional) – Range end

  • hi_4 (int or Tensor/Batch of int, optional) – Range end

  • hi_5 (int or Tensor/Batch of int, optional) – Range end

  • hi_6 (int or Tensor/Batch of int, optional) – Range end

  • hi_7 (int or Tensor/Batch of int, optional) – Range end

  • hi_8 (int or Tensor/Batch of int, optional) – Range end

  • hi_9 (int or Tensor/Batch of int, optional) – Range end

  • lo_0 (int or Tensor/Batch of int, optional) – Range start

  • lo_1 (int or Tensor/Batch of int, optional) – Range start

  • lo_10 (int or Tensor/Batch of int, optional) – Range start

  • lo_11 (int or Tensor/Batch of int, optional) – Range start

  • lo_12 (int or Tensor/Batch of int, optional) – Range start

  • lo_13 (int or Tensor/Batch of int, optional) – Range start

  • lo_14 (int or Tensor/Batch of int, optional) – Range start

  • lo_15 (int or Tensor/Batch of int, optional) – Range start

  • lo_16 (int or Tensor/Batch of int, optional) – Range start

  • lo_17 (int or Tensor/Batch of int, optional) – Range start

  • lo_18 (int or Tensor/Batch of int, optional) – Range start

  • lo_19 (int or Tensor/Batch of int, optional) – Range start

  • lo_2 (int or Tensor/Batch of int, optional) – Range start

  • lo_20 (int or Tensor/Batch of int, optional) – Range start

  • lo_21 (int or Tensor/Batch of int, optional) – Range start

  • lo_22 (int or Tensor/Batch of int, optional) – Range start

  • lo_23 (int or Tensor/Batch of int, optional) – Range start

  • lo_24 (int or Tensor/Batch of int, optional) – Range start

  • lo_25 (int or Tensor/Batch of int, optional) – Range start

  • lo_26 (int or Tensor/Batch of int, optional) – Range start

  • lo_27 (int or Tensor/Batch of int, optional) – Range start

  • lo_28 (int or Tensor/Batch of int, optional) – Range start

  • lo_29 (int or Tensor/Batch of int, optional) – Range start

  • lo_3 (int or Tensor/Batch of int, optional) – Range start

  • lo_30 (int or Tensor/Batch of int, optional) – Range start

  • lo_31 (int or Tensor/Batch of int, optional) – Range start

  • lo_4 (int or Tensor/Batch of int, optional) – Range start

  • lo_5 (int or Tensor/Batch of int, optional) – Range start

  • lo_6 (int or Tensor/Batch of int, optional) – Range start

  • lo_7 (int or Tensor/Batch of int, optional) – Range start

  • lo_8 (int or Tensor/Batch of int, optional) – Range start

  • lo_9 (int or Tensor/Batch of int, optional) – Range start

  • num_subscripts (int, optional) – Number of subscripts supplied, including full-range - used for validation only.

  • step_0 (int or Tensor/Batch of int, optional) – Range step

  • step_1 (int or Tensor/Batch of int, optional) – Range step

  • step_10 (int or Tensor/Batch of int, optional) – Range step

  • step_11 (int or Tensor/Batch of int, optional) – Range step

  • step_12 (int or Tensor/Batch of int, optional) – Range step

  • step_13 (int or Tensor/Batch of int, optional) – Range step

  • step_14 (int or Tensor/Batch of int, optional) – Range step

  • step_15 (int or Tensor/Batch of int, optional) – Range step

  • step_16 (int or Tensor/Batch of int, optional) – Range step

  • step_17 (int or Tensor/Batch of int, optional) – Range step

  • step_18 (int or Tensor/Batch of int, optional) – Range step

  • step_19 (int or Tensor/Batch of int, optional) – Range step

  • step_2 (int or Tensor/Batch of int, optional) – Range step

  • step_20 (int or Tensor/Batch of int, optional) – Range step

  • step_21 (int or Tensor/Batch of int, optional) – Range step

  • step_22 (int or Tensor/Batch of int, optional) – Range step

  • step_23 (int or Tensor/Batch of int, optional) – Range step

  • step_24 (int or Tensor/Batch of int, optional) – Range step

  • step_25 (int or Tensor/Batch of int, optional) – Range step

  • step_26 (int or Tensor/Batch of int, optional) – Range step

  • step_27 (int or Tensor/Batch of int, optional) – Range step

  • step_28 (int or Tensor/Batch of int, optional) – Range step

  • step_29 (int or Tensor/Batch of int, optional) – Range step

  • step_3 (int or Tensor/Batch of int, optional) – Range step

  • step_30 (int or Tensor/Batch of int, optional) – Range step

  • step_31 (int or Tensor/Batch of int, optional) – Range step

  • step_4 (int or Tensor/Batch of int, optional) – Range step

  • step_5 (int or Tensor/Batch of int, optional) – Range step

  • step_6 (int or Tensor/Batch of int, optional) – Range step

  • step_7 (int or Tensor/Batch of int, optional) – Range step

  • step_8 (int or Tensor/Batch of int, optional) – Range step

  • step_9 (int or Tensor/Batch of int, optional) – Range step

On this page