nvidia.dali.fn.zeros

nvidia.dali.fn.zeros(*, bytes_per_sample_hint=[0], dtype=DALIDataType.INT32, preserve=False, shape=None, device=None, name=None)

返回填充零的给定形状和类型的新数据。

支持的后端

• ‘cpu’

关键字参数:

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

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

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

• dtype (nvidia.dali.types.DALIDataType, 可选, 默认 = DALIDataType.INT32) – 输出数据类型。

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

• shape (int 或 list of int 或 TensorList of int, 可选) – 输出数据的形状。