jax.numpy.gcd#
- jax.numpy.gcd(x1, x2)[source]#
Compute the greatest common divisor of two arrays.
JAX implementation of
numpy.gcd()
.- Parameters:
x1 (ArrayLike) – First input array. The elements must have integer dtype.
x2 (ArrayLike) – Second input array. The elements must have integer dtype.
- Returns:
An array containing the greatest common divisors of the corresponding elements from the absolute values of x1 and x2.
- Return type:
See also
jax.numpy.lcm()
: compute the least common multiple of two arrays.
Examples
Scalar inputs:
>>> jnp.gcd(12, 18) Array(6, dtype=int32, weak_type=True)
Array inputs:
>>> x1 = jnp.array([12, 18, 24]) >>> x2 = jnp.array([5, 10, 15]) >>> jnp.gcd(x1, x2) Array([1, 2, 3], dtype=int32)
Broadcasting:
>>> x1 = jnp.array([12]) >>> x2 = jnp.array([6, 9, 12]) >>> jnp.gcd(x1, x2) Array([ 6, 3, 12], dtype=int32)