nvidia.dali.fn.ones

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

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

支持的后端

• ‘cpu’

关键字参数:

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

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

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

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

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

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