jax.numpy.lcm

jax.numpy.lcm(x1, x2)

计算两个数组的最小公倍数。

numpy.lcm() 的 JAX 实现。

参数:

• x1 (类似数组) – 第一个输入数组。元素必须具有整数 dtype。

• x2 (类似数组) – 第二个输入数组。元素必须具有整数 dtype。

返回:

一个数组，包含来自 x1 和 x2 的绝对值的相应元素的最小公倍数。

返回类型:

数组

另请参阅

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