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, optional, default = [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, optional, default = False) – 阻止操作符从图中删除，即使其输出未使用。

在本页中