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) – 即使其输出未使用，也防止运算符从图中删除。

在此页