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)

回傳型別：

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)