jax.numpy.gcd

内容

jax.numpy.gcd#

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

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

JAX 实现 numpy.gcd()。

参数:

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

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

返回:

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

返回类型:

Array

参见

示例

标量输入

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