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 ) – 布尔值，可选，默认值为 True。如果为 True，则在结果中包含 stop 值。如果为 False，则排除 stop 值。

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

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