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, 可选, 默认值 = [0]) –

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

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

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

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

在此页上