jax.lax.neg

jax.lax.neg(x)

元素级取反：\(-x\)。

此函数直接降低到 stablehlo.negate 操作。

参数:

x (ArrayLike) – 输入数组

返回值:

与 x 形状和数据类型相同的数组，包含元素级负数。

返回类型:

数组

Notes

对于无符号整数输入，此函数返回 2 ** nbits - x，其中 nbits 是整数表示中的位数。