jax.numpy.copysign

jax.numpy.copysign(x1, x2, /)

将 x2 中每个元素的符号复制到 x1 中对应的元素。

JAX implementation of numpy.copysign.

参数:

• x1 (ArrayLike) – 输入数组

• x2 (ArrayLike) – 用于确定符号的数组的元素必须与 x1 广播兼容。

返回:

包含 x1 中可能更改的元素的数组对象，始终提升为不精确的数据类型，并且形状为 jnp.broadcast_shapes(x1.shape, x2.shape)

返回类型:

Array

示例

>>> x1 = jnp.array([5, 2, 0])
>>> x2 = -1
>>> jnp.copysign(x1, x2)
Array([-5., -2., -0.], dtype=float32)

>>> x1 = jnp.array([6, 8, 0])
>>> x2 = 2
>>> jnp.copysign(x1, x2)
Array([6., 8., 0.], dtype=float32)

>>> x1 = jnp.array([2, -3])
>>> x2 = jnp.array([[1],[-4], [5]])
>>> jnp.copysign(x1, x2)
Array([[ 2.,  3.],
       [-2., -3.],
       [ 2.,  3.]], dtype=float32)