jax.numpy.lcm

内容

jax.numpy.lcm#

jax.numpy.lcm(x1, x2)[source]#

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

JAX 实现 numpy.lcm()。

参数：

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

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

返回值：

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

返回类型：

数组

参见

示例

标量输入

>>> 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)
内容