jax.nn.squareplus# jax.nn.squareplus(x, b=4)[源代码][源代码]# Squareplus 激活函数。 计算逐元素函数 \[\mathrm{squareplus}(x) = \frac{x + \sqrt{x^2 + b}}{2}\] 如 https://arxiv.org/abs/2112.11687 所述。 参数: x (ArrayLike) – 输入数组 b (ArrayLike) – 平滑参数 返回类型: Array
