nvidia.dali.fn.ones_like

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

返回与输入数组具有相同形状和类型的新数据，并用 1 填充。

支持的后端

• ‘cpu’

参数:

__data_like (TensorList) – 要从中复制形状和类型的输入数据值。

关键字参数:

• bytes_per_sample_hint (int 或 list of int, optional, default = [0]) –

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

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

• dtype (nvidia.dali.types.DALIDataType, optional, default = DALIDataType.INT32) – 覆盖输出数据类型。

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