jax.numpy.copysign#

jax.numpy.copysign(x1, x2, /)[源代码]#

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

JAX 对 numpy.copysign 的实现。

参数:

  • x1 (ArrayLike) – 输入数组

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

返回:

一个数组对象，包含 x1 可能已更改的元素，始终提升为非精确（inexact）数据类型，并且形状为 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)