jax.numpy.gcd

jax.numpy.gcd(x1, x2)

計算兩個陣列的最大公約數。

JAX 實作的 numpy.gcd()。

參數:

• x1 (ArrayLike) – 第一個輸入陣列。元素必須為整數 dtype。

• x2 (ArrayLike) – 第二個輸入陣列。元素必須為整數 dtype。

回傳:

一個陣列，包含 x1 和 x2 絕對值對應元素的最大公約數。

回傳類型:

Array

參見

• 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)