nvidia.dali.fn.coin_flip#

nvidia.dali.fn.coin_flip(__shape_like=None, /, *, bytes_per_sample_hint=[0], dtype=None, preserve=False, probability=0.5, seed=-1, shape=None, device=None, name=None)#

警告

此运算符已弃用。请改用 random.coin_flip()

生成遵循伯努利分布的随机布尔值。

生成值 1 (true) 的概率由 probability 参数确定。

生成数据的形状可以通过 shape 参数显式指定,或者选择与输入形状匹配(如果提供输入)。如果两者都不存在,则每个样本生成一个值。

支持的后端
  • ‘cpu’

  • ‘gpu’

参数:

__shape_like (TensorList, 可选) – 如果提供此输入,则其形状将用于推断输出的形状。

关键字参数:
  • bytes_per_sample_hint (int 或 list of int, 可选, 默认 = [0]) –

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

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

  • dtype (nvidia.dali.types.DALIDataType, 可选) –

    输出数据类型。

    注意

    生成的数字将转换为输出数据类型,并在必要时进行舍入和钳位。

  • preserve (bool, 可选, 默认 = False) – 即使运算符的输出未使用,也阻止从图中删除该运算符。

  • probability (float 或 TensorList of float, 可选, 默认 = 0.5) – 值 1 的概率。

  • seed (int, 可选, 默认 = -1) – 随机种子;如果未设置,将自动分配一个种子。

  • shape (intlist of intTensorList of int, 可选) – 输出数据的形状。