jax.lax.logistic

jax.lax.logistic(x, accuracy=None)

逐元素 logistic（sigmoid）函数：\(\frac{1}{1 + e^{-x}}\)。

没有 HLO logistic/sigmoid 原语，因此这会降低为一系列 HLO 算术运算。

参数:

x (ArrayLike) – 输入数组。必须具有浮点或复数 dtype。

返回:

与 x 形状和 dtype 相同的数组，包含逐元素的 logistic/sigmoid 函数。

返回类型:

数组

另请参阅

• jax.nn.sigmoid()：此功能的替代 API。