zeros_cn.rst 1.5 KB
Newer Older
1 2
.. _cn_api_tensor_zeros:

T
tianshuo78520a 已提交
3 4
zeros
-------------------------------
5

6
.. py:function:: paddle.zeros(shape, dtype=None, name=None)
S
swtkiwi 已提交
7 8 9



10 11 12
该OP创建形状为 ``shape`` 、数据类型为 ``dtype`` 且值全为0的Tensor。

参数:
13
    - **shape** (tuple|list|Tensor) - 输出Tensor的形状, ``shape`` 的数据类型为int32或者int64。
14 15
    - **dtype** (np.dtype|core.VarDesc.VarType|str,可选) - 输出Tensor的数据类型,数据类型必须为bool、float16、float32、float64、int32或int64。若为None,数据类型为float32, 默认为None。
    - **name** (str, 可选) - 输出的名字。一般无需设置,默认值为None。该参数供开发人员打印调试信息时使用,具体用法请参见 :ref:`api_guide_Name` 。
16 17 18

返回:值全为0的Tensor,数据类型和 ``dtype`` 定义的类型一致。

19 20 21
抛出异常:
    - ``TypeError`` - 当 ``dtype`` 不是bool、 float16、float32、float64、int32、int64和None时。
    - ``TypeError`` - 当 ``shape`` 不是tuple、list、或者Tensor时, 当 ``shape`` 为Tensor,其数据类型不是int32或者int64时。
22 23 24 25 26 27

**代码示例**:

.. code-block:: python

    import paddle
28 29 30 31 32 33 34 35 36 37 38 39
    paddle.enable_imperative()  # Now we are in imperative mode
    data = paddle.zeros(shape=[3, 2], dtype='float32') 
    # [[0. 0.]
    #  [0. 0.]
    #  [0. 0.]]
    
    data = paddle.zeros(shape=[2, 2]) 
    # [[0. 0.]
    #  [0. 0.]]
    
    # shape is a Tensor
    shape = paddle.fill_constant(shape=[2], dtype='int32', value=2)
40
    data3 = paddle.zeros(shape=shape, dtype='int32') 
41 42
    # [[0 0]
    #  [0 0]]
43