伯努利

伯努利 - 22

版本

  • 名称: Bernoulli (GitHub)

  • 域名: main

  • since_version: 22

  • function: True

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本22起可用。

摘要

从伯努利分布中抽取二进制随机数（0或1）。输入张量应包含用于抽取二进制随机数的概率p（范围在[0,1]内的值），其中输出1的概率为p，输出0的概率为(1-p)。

此操作符是非确定性的，在不同的实现中可能不会产生相同的值（即使指定了种子）。

属性

  • dtype - INT :

    输出张量元素的数据类型。如果未指定，我们将使用输入张量的数据类型。

  • seed - FLOAT :

    （可选）随机生成器的种子，如果未指定，我们将自动生成一个。

输入

  • 输入 (异构) - T1:

    输入中的所有值必须在范围：[0, 1]内。

输出

  • 输出 (异构) - T2:

    返回的输出张量只有0或1的值，形状与输入张量相同。

类型约束

  • T1 在 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) ) 中:

    将输入类型限制为浮点张量。

  • T2 在 ( tensor(bfloat16), tensor(bool), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) ):

    将输出类型限制为所有数值张量和布尔张量。

伯努利 - 15

版本

  • 名称: Bernoulli (GitHub)

  • 域名: main

  • since_version: 15

  • function: True

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本15起可用。

摘要

从伯努利分布中抽取二进制随机数（0或1）。输入张量应包含用于抽取二进制随机数的概率p（范围在[0,1]内的值），其中输出1的概率为p，输出0的概率为(1-p)。

此操作符是非确定性的，在不同的实现中可能不会产生相同的值（即使指定了种子）。

属性

  • dtype - INT :

    输出张量元素的数据类型。如果未指定，我们将使用输入张量的数据类型。

  • seed - FLOAT :

    （可选）随机生成器的种子，如果未指定，我们将自动生成一个。

输入

  • 输入 (异构) - T1:

    输入中的所有值必须在范围：[0, 1]内。

输出

  • 输出 (异构) - T2:

    返回的输出张量仅包含0或1的值，与输入张量的形状相同。

类型约束

  • T1 在 ( tensor(double), tensor(float), tensor(float16) ) 中:

    将输入类型限制为浮点张量。

  • T2 在 ( tensor(bfloat16), tensor(bool), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) ):

    将输出类型限制为所有数值张量和布尔张量。