linspace_cn.rst 2.0 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5
.. _cn_api_fluid_layers_linspace:

linspace
-------------------------------

6
.. py:function:: paddle.fluid.layers.linspace(start, stop, num, dtype=None, name=None)
H
Hao Wang 已提交
7

8 9
该OP返回一个Tensor,Tensor的值为在区间start和stop上均匀间隔的num个值,输出Tensor的长度为num。
**注意:该OP不进行梯度计算**
H
Hao Wang 已提交
10 11
 
参数:
W
wangchaochaohu 已提交
12 13
    - **start** (int|float|Tensor) – ``start`` 是区间开始的变量,可以是一个浮点标量,或是一个shape为[1]的Tensor,该Tensor的数据类型可以是float32,float64,int32 或者int64。
    - **stop** (int|float|Tensor) – ``stop`` 是区间结束的变量,可以是一个浮点标量,或是一个shape为[1]的Tensor,该Tensor的数据类型可以是float32,float64,int32或者int64。
14
    - **num** (int|Tensor) – ``num`` 是给定区间内需要划分的区间数,可以是一个整型标量,或是一个shape为[1]的Tensor,该Tensor的数据类型需为int32。
W
wangchaochaohu 已提交
15
    - **dtype** (np.dtype|str, 可选) – 输出Tensor的数据类型,可以是float32,float64, int32或者int64。如果dtype的数据类型为None,输出Tensor数据类型为float32。
16
    - **name** (str, 可选)- 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。
H
Hao Wang 已提交
17

18
返回:表示等间隔划分结果的1-D Tensor,该Tensor的shape大小为 :math:`[num]` ,在mum为1的情况下,仅返回包含start元素值的Tensor。
H
Hao Wang 已提交
19

20
抛出异常:
W
wangchaochaohu 已提交
21 22 23 24
    - ``TypeError`` - 当 ``start`` 或者 ``stop`` 的数据类型不是float32,float64, int32或者int64。
    - ``TypeError`` - 当 ``dtype`` 的类型不是float32,float64,int32或者int64。
    - ``TypeError`` - ``num`` 的类型必须是int当 ``num`` 不是Tensor的时候。
    - ``TypeError`` - ``num`` 的数据类型必须是int32当 ``num`` 是Tensor的时候。
H
Hao Wang 已提交
25 26 27 28 29 30 31 32 33 34 35 36 37

**代码示例**:

.. code-block:: python

      import paddle.fluid as fluid
      data = fluid.layers.linspace(0, 10, 5, 'float32') # [0.0,  2.5,  5.0,  7.5, 10.0]
      data = fluid.layers.linspace(0, 10, 1, 'float32') # [0.0]