jax.lax.logistic#

jax.lax.logistic(x, accuracy=None)[source]#

Elementwise logistic (sigmoid) 函数：$\frac{1}{1+e^{-x}}$。

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

参数:

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

返回值:

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

返回类型:

数组

另请参阅

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