nvidia.dali.fn.element_extract#

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

从输入序列中提取一个或多个元素。

输出是输入的第一（最外层）维度中的切片。输出的数量与 element_map 中提供的元素数量相同。

例如，对于 element_map = [2, 0, 3]，将有三个输出，分别包含输入序列的第 2、第 0 和第 3 个元素。

如果提供了输入布局，则必须以 F 维度开头。输出的维度将比输入少一个，也就是说，对于 FHWC 输入，输出将是 HWC 元素。

此运算符期望序列输入。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__input (TensorList) – 运算符的输入。

关键字参数:

  • bytes_per_sample_hint (int 或 list of int, optional, default = [0]) –

    每个样本的输出大小提示（以字节为单位）。

    如果指定，则位于 GPU 或页锁定主机内存中的运算符输出将预先分配，以容纳此大小的样本批次。

  • element_map (int 或 list of int) – 要提取的元素的索引。

  • preserve (bool, optional, default = False) – 即使运算符的输出未使用，也阻止将其从图中移除。

在此页上