jax.numpy.true_divide

内容

jax.numpy.true_divide#

jax.numpy.true_divide(x1, x2, /)[source]#

按元素计算 x1 除以 x2。

JAX 实现 numpy.true_divide()。

参数：

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

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

返回值：

包含按元素商的数组，将始终使用浮点除法。

返回类型：

Array

示例

>>> 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() 用于整数除法

内容