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 的 int, 可选) – 输出数据的形状。