jax.numpy.geomspace

jax.numpy.geomspace(start, stop, num=50, endpoint=True, dtype=None, axis=0)

生成几何级数的值。

JAX 实现的 numpy.geomspace()。

参数:

• start (ArrayLike) – 标量或数组。指定起始值。

• stop (ArrayLike) – 标量或数组。指定终止值。

• num (int) – int，可选，默认值=50。要生成的值的数量。

• endpoint (bool) – bool，可选，默认值=True。如果为True，则在结果中包含 stop 值。如果为False，则排除 stop 值。

• dtype (DTypeLike | None | None) – 可选。指定输出的数据类型。

• axis (int) – int，可选，默认值=0。生成几何级数所沿的轴。

返回值:

一个包含几何级数值的数组。

返回类型:

数组

另请参阅

• jax.numpy.arange()：生成给定起始点和步长的 N 个等间隔值。

• jax.numpy.linspace()：生成等间隔值。

• jax.numpy.logspace()：生成对数间隔值。

示例

列出 1 到 16 之间 5 个几何间隔的值

>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.geomspace(1, 16, 5)
Array([ 1.,  2.,  4.,  8., 16.], dtype=float32)

列出 1 到 16 之间 4 个几何间隔的值，且 endpoint=False

>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.geomspace(1, 16, 4, endpoint=False)
Array([1., 2., 4., 8.], dtype=float32)

多维几何级数

>>> start = jnp.array([1, 1000])
>>> stop = jnp.array([27, 1])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.geomspace(start, stop, 4)
Array([[   1., 1000.],
       [   3.,  100.],
       [   9.,   10.],
       [  27.,    1.]], dtype=float32)