jax.lax.sign

jax.lax.sign(x)

逐元素符号函数。

此函数直接降低为 stablehlo.sign 操作。

参数:

x (ArrayLike) – 输入数组

返回值:

与 x 形状和 dtype 相同的数组，包含值的符号，如下面的“说明”中所定义。

返回类型:

Array

Notes

对于浮点输入，返回

\[\begin{split} \mathrm{sign}(x) = \begin{cases} -1 & x < 0\\ -0 & x = -0\\ \mathit{NaN} & x = \mathit{NaN}\\ +0 & x = +0\\ 1 & x > 0 \end{cases}\end{split}\]

对于有符号整数输入，返回

\[\begin{split}\mathrm{sign}(x) = \begin{cases} -1 & x < 0\\ 0 & x = 0\\ 1 & x > 0 \end{cases}\end{split}\]

对于复数输入，返回复数相位，即 \(\mathrm{sign}(x) = x / |x|\)。