jax.numpy.positive

内容

jax.numpy.positive#

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

返回输入的逐元素正值。

JAX 实现 numpy.positive.

参数:

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

返回值:

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

返回类型:

数组

注意

jnp.positive 等价于 x.copy(),并且仅针对支持算术运算的类型定义。

参见

示例

对于实数值输入

>>> x = jnp.array([-5, 4, 7., -9.5])
>>> jnp.positive(x)
Array([-5. ,  4. ,  7. , -9.5], dtype=float32)
>>> x.copy()
Array([-5. ,  4. ,  7. , -9.5], dtype=float32)

对于复数输入

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

对于 uint32

>>> x2 = jnp.array([6, 0, -4]).astype(jnp.uint32)
>>> x2
Array([         6,          0, 4294967292], dtype=uint32)
>>> jnp.positive(x2)
Array([         6,          0, 4294967292], dtype=uint32)