jax.lax.sign

jax.lax.sign(x)

元素级符号。

对于浮点输入，返回 \(\mathrm{sign}(x) = \begin{cases} -1 & x < 0\\ -0 & x = -0\\ \mathit{NaN} & x = \mathit{NaN}\\ +0 & x = +0\\ 1 & x > 0 \end{cases}\)

对于有符号整数输入，返回 \(\mathrm{sign}(x) = \begin{cases} -1 & x < 0\\ 0 & x = 0\\ 1 & x > 0 \end{cases}\)

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

参数：

x (ArrayLike)

返回类型：

数组