nvidia.dali.fn.shapes#

nvidia.dali.fn.shapes(__input, /, *, bytes_per_sample_hint=[0], dtype=DALIDataType.INT64, preserve=False, device=None, name=None)#

警告

此运算符现已弃用。

请使用 nvidia.dali.pipeline.DataNode.shape() 代替。

返回输入批次中张量的形状。

此运算符允许序列输入并支持体积数据。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

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

关键字参数:

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

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

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

  • dtype (nvidia.dali.types.DALIDataType, 可选, 默认值 = DALIDataType.INT64) – 大小将转换为的数据类型。

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

  • type (nvidia.dali.types.DALIDataType) –

    警告

    参数 type 是 dtype 的已弃用别名。请使用 dtype 代替。

在本页中