nvidia.dali.fn.sequence_rearrange#

nvidia.dali.fn.sequence_rearrange(__input, /, *, bytes_per_sample_hint=[0], new_order, preserve=False, device=None, name=None)#

重新排列序列中的帧。

假设最外层维度表示序列中的帧索引。如果输入具有非空布局描述，则它必须以 F (帧) 开头。

此操作器允许序列输入。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__input (TensorList) – 操作器的输入。

关键字参数:

  • bytes_per_sample_hint (int 或 list of int, 可选, 默认 = [0]) –

    输出大小提示，以每样本字节为单位。

    如果指定，则驻留在 GPU 或分页锁定主机内存中的操作器的输出将被预先分配以适应此大小的样本批次。

  • new_order (int 或 list of int 或 TensorList of int) –

    描述每个样本中元素新顺序的列表。

    位置 i 的输出序列将包含来自输入序列的元素 new_order[i]

    out[i, ...] = in[new_order[i], ...]

    元素可以重复或删除，但不允许空输出序列。只允许在 new_order 中使用 [0, input_outermost_extent) 中的索引。可以为每个样本指定为 1D 张量。

  • preserve (bool, 可选, 默认 = False) – 阻止操作器从图中移除，即使其输出未使用。

在本页中