jax.numpy.geomspace

内容

jax.numpy.geomspace#

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

返回对数刻度上均匀间隔的数字（几何级数）。

LAX 后端实现 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, 可选) – 结果中存储样本的轴。仅当 start 或 stop 为数组类型时才相关。默认情况下 (0)，样本将沿插入开头的新轴排列。使用 -1 可在末尾获取一个轴。

返回:

samples – num 个样本，在对数刻度上等间距。

返回类型:

ndarray

内容