nvidia.dali.fn.permute_batch

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

返回一个批次的张量，该张量通过基于 indices 参数中给出的索引从输入中选择张量来构建

out_tensor[i] = in_tensor[indices[i]]

支持的后端

• ‘cpu’

• ‘gpu’

参数:

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

关键词参数:

• bytes_per_sample_hint (int 或 int 列表, 可选, 默认值 = [0]) –

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

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

• indices (int 或 int 列表 或 TensorList 的 int) –

  索引列表，与当前批次大小匹配，或表示输入批次中张量索引的标量批次。

  索引必须在 [0..batch_size) 范围内。 允许重复和遗漏。

• preserve (bool, 可选, 默认值 = False) – 即使操作符的输出未使用，也阻止将其从图中删除。