jax.numpy.hypot#

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

傳回直角三角形給定兩股的元素級斜邊。

numpy.hypot 的 JAX 實作。

參數：

  • x1 (ArrayLike) – 純量或陣列。指定直角三角形的其中一股。complex dtype 不支援。

  • x2 (ArrayLike) – 純量或陣列。指定直角三角形的另一股。complex dtype 不支援。x1 和 x2 必須具有相同的形狀或可廣播相容。

傳回：

一個陣列，包含直角三角形給定兩股 x1 和 x2 的斜邊，並提升為非精確 dtype。

傳回型別：

Array

註解

jnp.hypot 是計算 jnp.sqrt(x1 ** 2 + x2 **2) 更數值穩定的方法。

範例

>>> jnp.hypot(3, 4)
Array(5., dtype=float32, weak_type=True)
>>> x1 = jnp.array([[3, -2, 5],
...                 [9, 1, -4]])
>>> x2 = jnp.array([-5, 6, 8])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.hypot(x1, x2)
Array([[ 5.831,  6.325,  9.434],
       [10.296,  6.083,  8.944]], dtype=float32)