jax.numpy.gcd

jax.numpy.gcd(x1, x2)

计算两个数组的最大公约数。

numpy.gcd() 的 JAX 实现。

参数:

• x1 (ArrayLike) – 第一个输入数组。元素必须为整数数据类型。

• x2 (ArrayLike) – 第二个输入数组。元素必须为整数数据类型。

返回值:

一个数组，包含 x1 和 x2 的绝对值中对应元素的最大公约数。

返回类型:

数组

另请参阅

• jax.numpy.lcm(): 计算两个数组的最小公倍数。

示例

标量输入

>>> jnp.gcd(12, 18)
Array(6, dtype=int32, weak_type=True)

数组输入

>>> x1 = jnp.array([12, 18, 24])
>>> x2 = jnp.array([5, 10, 15])
>>> jnp.gcd(x1, x2)
Array([1, 2, 3], dtype=int32)

广播

>>> x1 = jnp.array([12])
>>> x2 = jnp.array([6, 9, 12])
>>> jnp.gcd(x1, x2)
Array([ 6,  3, 12], dtype=int32)