jax.numpy.hypot

jax.numpy.hypot(x1, x2, /)

Given the “legs” of a right triangle, return its hypotenuse.

LAX-backend implementation of numpy.hypot().

Original docstring below.

Equivalent to sqrt(x1**2 + x2**2), element-wise. If x1 or x2 is scalar_like (i.e., unambiguously cast-able to a scalar type), it is broadcast for use with each element of the other argument. (See Examples)

Parameters:

• x1 (array_like) – Leg of the triangle(s). If x1.shape != x2.shape, they must be broadcastable to a common shape (which becomes the shape of the output).

• x2 (array_like) – Leg of the triangle(s). If x1.shape != x2.shape, they must be broadcastable to a common shape (which becomes the shape of the output).

Returns:

z – The hypotenuse of the triangle(s). This is a scalar if both x1 and x2 are scalars.

Return type:

ndarray