jax.numpy.copysign#

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

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

numpy.copysign 的 JAX 实现。

参数:

  • x1 (ArrayLike) – 输入数组

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

返回:

一个数组对象，包含 x1 中可能已更改的元素，总是提升为非精确的 dtype，形状为 jnp.broadcast_shapes(x1.shape, x2.shape)

返回类型:

数组

示例

>>> 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)