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) – 選用。指定輸出的 dtype。

• axis (int) – int，選用，預設值=0。產生 geomspace 的軸。

回傳值:

包含幾何級數間隔值的陣列。

回傳型別:

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)

多維 geomspace

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