伯努利¶
伯努利 - 22¶
版本¶
名称: 伯努利 (GitHub)
域:
main自版本:
22函数:
True支持级别:
SupportType.COMMON形状推断:
True
此版本的运算符已自版本 22起可用。
摘要¶
从伯努利分布中抽取二进制随机数（0 或 1）。输入张量应包含用于抽取二进制随机数的概率 p（[0,1] 范围内的值），其中以概率 p 生成输出 1，并以概率（1-p）生成输出 0。
此运算符是非确定性的，并且可能在不同的实现中产生不同的值（即使指定了种子）。
属性¶
dtype - INT :
输出张量元素的数据类型。如果未指定，我们将使用输入张量的数据类型。
seed - FLOAT :
（可选）随机生成器的种子，如果未指定，我们将自动生成一个。
输入¶
input (异构) - T1
输入中的所有值都必须在以下范围内：[0, 1]。
输出¶
output (异构) - T2
返回的输出张量仅包含值 0 或 1，与输入张量形状相同。
类型约束¶
T1 in (
tensor(bfloat16),tensor(double),tensor(float),tensor(float16))将输入类型约束为浮点张量。
T2 in (
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¶
版本¶
名称: 伯努利 (GitHub)
域:
main自版本:
15函数:
True支持级别:
SupportType.COMMON形状推断:
True
此版本的运算符已自版本 15起可用。
摘要¶
从伯努利分布中抽取二进制随机数（0 或 1）。输入张量应包含用于抽取二进制随机数的概率 p（[0,1] 范围内的值），其中以概率 p 生成输出 1，并以概率（1-p）生成输出 0。
此运算符是非确定性的，并且可能在不同的实现中产生不同的值（即使指定了种子）。
属性¶
dtype - INT :
输出张量元素的数据类型。如果未指定，我们将使用输入张量的数据类型。
seed - FLOAT :
（可选）随机生成器的种子，如果未指定，我们将自动生成一个。
输入¶
input (异构) - T1
输入中的所有值都必须在以下范围内：[0, 1]。
输出¶
output (异构) - T2
返回的输出张量仅包含值 0 或 1，与输入张量形状相同。
类型约束¶
T1 in (
tensor(double),tensor(float),tensor(float16))将输入类型约束为浮点张量。
T2 in (
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))将输出类型约束为所有数值张量和布尔张量。