linspace¶
返回一个 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.]