jax.numpy.lcm

jax.numpy.lcm(x1, x2)

計算兩個陣列的最小公倍數。

numpy.lcm() 的 JAX 實作。

參數:

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

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

回傳:

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

回傳型別:

Array

參見

• jax.numpy.gcd()：計算兩個陣列的最大公因數。

範例

純量輸入

>>> jnp.lcm(12, 18)
Array(36, dtype=int32, weak_type=True)

陣列輸入

>>> x1 = jnp.array([12, 18, 24])
>>> x2 = jnp.array([5, 10, 15])
>>> jnp.lcm(x1, x2)
Array([ 60,  90, 120], dtype=int32)

廣播

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