jax.numpy.negative

内容

jax.numpy.negative#

jax.numpy.negative(x, /)[source]#

返回输入的逐元素负值。

JAX 实现 numpy.negative

参数:

x (ArrayLike) – 输入数组或标量。

返回值:

一个与 x 形状和数据类型相同的数组,包含 -x

返回类型:

数组

参见

注意

jnp.negative 应用于 无符号整数 时,会产生其二进制补码负数的结果,由于整数下溢,通常会导致意外的大正数值。

示例

对于实数值输入

>>> x = jnp.array([0., -3., 7])
>>> jnp.negative(x)
Array([-0.,  3., -7.], dtype=float32)

对于复数输入

>>> x1 = jnp.array([1-2j, -3+4j, 5-6j])
>>> jnp.negative(x1)
Array([-1.+2.j,  3.-4.j, -5.+6.j], dtype=complex64)

对于 unit32

>>> x2 = jnp.array([5, 0, -7]).astype(jnp.uint32)
>>> x2
Array([         5,          0, 4294967289], dtype=uint32)
>>> jnp.negative(x2)
Array([4294967291,          0,          7], dtype=uint32)