jax.numpy.geomspace

目录

jax.numpy.geomspace#

jax.numpy.geomspace(start, stop, num=50, endpoint=True, dtype=None, axis=0)[源代码][源代码]#

返回在对数尺度上均匀分布的数字（几何级数）。

LAX-backend 对 numpy.geomspace() 的实现。

原始文档字符串如下。

这与 logspace 类似，但直接指定了端点。每个输出样本是前一个的常数倍。

在 1.16.0 版本发生变更: 现在支持非标量的 start 和 stop。

参数:

  • start (array_like) – 序列的起始值。

  • stop (array_like) – 序列的最终值，除非 endpoint 为 False。在这种情况下，num + 1 个值在日志空间中分布在区间上，其中除了最后一个（长度为 num 的序列）之外的所有值都被返回。

  • num (integer, optional) – 要生成的样本数量。默认是 50。

  • endpoint (boolean, optional) – 如果为真，stop 是最后一个样本。否则，它不包括在内。默认为 True。

  • dtype (dtype) – 输出数组的类型。如果未指定 dtype，则数据类型会根据 start 和 stop 推断。推断的 dtype 永远不会是整数；即使参数会产生整数数组，也会选择 float。

  • axis (int, optional) – 结果中存储样本的轴。仅在 start 或 stop 为类数组时相关。默认情况下（0），样本将沿插入到开始位置的新轴排列。使用 -1 以在末尾获取一个轴。

返回:

样本 – num 个样本，在对数尺度上等间隔分布。

返回类型:

ndarray

目录