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, optional, default = [0]) –

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

    如果指定，则将预先分配驻留在 GPU 或分页锁定主机内存中的操作符输出，以适应此大小的样本批次。

  • key (str) –

    指定请求哪个属性。

    支持以下属性

    • "source_info": 返回类型：字节数组。

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

    • "layout": 返回类型：字节数组

      给定张量中的 数据布局。

  • preserve (bool, optional, default = False) – 即使操作符的输出未使用，也阻止从图中删除该操作符。

在本页