linspace

paddle. linspace ( start, stop, num, dtype=None, name=None ) [源代码]

返回一个 Tensor,Tensor 的值为在区间 start 和 stop 上均匀间隔的 num 个值,输出 Tensor 的长度为 num。 注意:不进行梯度计算

参数

  • start (int|float|Tensor) – start 是区间开始的变量,可以是一个 int、float,或是一个 shape 为[0]的 Tensor,该 Tensor 的数据类型可以是 int32,int64,float32,float64。

  • stop (int|float|Tensor) – stop 是区间结束的变量,可以是一个 int、float,或是一个 shape 为[0]的 Tensor,该 Tensor 的数据类型可以是 int32,int64,float32,float64。

  • num (int|Tensor) – num 是给定区间内需要划分的区间数,可以是一个 int,或是一个 shape 为[0]的 Tensor,该 Tensor 的数据类型需为 int32。

  • dtype (np.dtype|str,可选) – 输出 Tensor 的数据类型,可以是 int32,int64,float32,float64。如果 dtype 的数据类型为 None,输出 Tensor 数据类型为 float32。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

表示等间隔划分结果的 1-D Tensor,该 Tensor 的 shape 大小为 \([num]\),在 num 为 1 的情况下,仅返回包含 start 元素值的 Tensor。

代码示例

>>> import paddle
>>> data = paddle.linspace(0, 10, 5, 'float32')
>>> print(data.numpy())
[0. 2.5 5. 7.5 10.]
>>> data = paddle.linspace(0, 10, 1, 'float32')
>>> print(data.numpy())
[0.]

使用本API的教程文档