jax.numpy.geomspace

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

生成几何等比数值。

JAX 对 numpy.geomspace() 的实现。

参数:

• start (Array | ndarray | bool | number | bool | int | float | complex) – 标量或数组。指定起始值。

• stop (Array | ndarray | bool | number | bool | int | float | complex) – 标量或数组。指定结束值。

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

• endpoint (bool) – 布尔型，可选，默认值=True。如果为True，则结果中包含 stop 值。如果为False，则不包含 stop 值。

• dtype (str | type[Any] | dtype | SupportsDType | None) – 可选。指定输出的数据类型 (dtype)。

• axis (int) – 整型，可选，默认值=0。生成几何等比数值的轴。

返回:

包含几何等比数值的数组。

返回类型:

Array

另请参阅

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