jax.numpy.true_divide#

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

计算 x1 除以 x2 的逐元素除法。

JAX 对 numpy.true_divide() 的实现。

参数:

  • x1 (ArrayLike) – 输入数组，被除数

  • x2 (ArrayLike) – 输入数组，除数

返回值:

一个包含逐元素商的数组，始终使用浮点除法。

返回类型:

数组

示例

>>> x1 = jnp.array([3, 4, 5])
>>> x2 = 2
>>> jnp.true_divide(x1, x2)
Array([1.5, 2. , 2.5], dtype=float32)

>>> x1 = 24
>>> x2 = jnp.array([3, 4, 6j])
>>> jnp.true_divide(x1, x2)
Array([8.+0.j, 6.+0.j, 0.-4.j], dtype=complex64)

>>> x1 = jnp.array([1j, 9+5j, -4+2j])
>>> x2 = 3j
>>> jnp.true_divide(x1, x2)
Array([0.33333334+0.j       , 1.6666666 -3.j       ,
       0.6666667 +1.3333334j], dtype=complex64)

另请参阅

jax.numpy.floor_divide() 用于整数除法