jax.numpy.hypot#

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

返回直角三角形给定两直角边的逐元素斜边。

JAX 对 numpy.hypot 的实现。

参数:

  • x1 (ArrayLike) – 标量或数组。指定直角三角形的一条直角边。不支持 complex 数据类型。

  • x2 (ArrayLike) – 标量或数组。指定直角三角形的另一条直角边。不支持 complex 数据类型。x1 和 x2 必须具有相同的形状或广播兼容。

返回:

一个数组，其中包含给定直角边 x1 和 x2 的直角三角形的斜边，并提升为非精确数据类型。

返回类型:

数组

注意

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)