nvidia.dali.fn.get_property#

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

返回作为输入传递的张量的属性。

输出的类型将取决于所请求属性的 key。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__input (TensorList) – 算子的输入。

关键词参数:

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

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

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

  • key (str) –

    指定请求哪个属性。

    支持以下属性

    • "source_info": 返回类型：byte-array。

      类似字符串的字节数组，其中包含有关样本来源的信息。 例如，在通过 fn.readers.file() 加载的张量上调用的 fn.get_property() 返回张量来源文件的完整路径。

    • "layout": 返回类型：byte-array

      给定张量中的数据布局。

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

本页内容