jax.numpy.positive

目录

jax.numpy.positive#

jax.numpy.positive(x, /)[源代码][源代码]#

返回输入元素的正值。

JAX implementation of numpy.positive.

参数:

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

返回:

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

返回类型:

Array

备注

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)