jax.numpy.gcd

Contents

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:

Array

See also

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)