ConstantPad2d_cn.rst 1.2 KB
Newer Older
L
littletomatodonkey 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
.. _cn_api_nn_ConstantPad2d:

ConstantPad2d
-------------------------------
.. py:class:: paddle.nn.ConstantPad2d(padding, value=0.0, data_format="NCHW", name=None)

**ConstantPad2d**

按照 padding 对输入 以constant模式进行 ``pad``,即填充固定值。

参数:
  - **padding** (Tensor | List[int32]) - 填充大小。pad的格式为[pad_left, pad_right, pad_top, pad_bottom]。
  - **value** (float32) - 待填充的值,默认值为0.0。
  - **data_format** (str)  - 指定input的format,可为 `'NCHW'` 或者 `'NHWC'`,默认值为`'NCHW'`。
  - **name** (str, 可选) - 该参数供开发人员打印调试信息时使用,具体用法请参见 :ref:`api_guide_Name` ,缺省值为None。

返回:无

**代码示例**

..  code-block:: python

    import paddle
    import paddle.nn as nn
    import numpy as np
    paddle.disable_static()

    input_shape = (1, 1, 2, 3)
    pad = [1, 0, 1, 2]
    data = np.arange(np.prod(input_shape), dtype=np.float32).reshape(input_shape) + 1
    my_pad = nn.ConstantPad2d(padding=pad)
    data = paddle.to_tensor(data)
    result = my_pad(data)
    print(result.numpy())
    # [[[[0. 0. 0. 0.]
    #    [0. 1. 2. 3.]
    #    [0. 4. 5. 6.]
    #    [0. 0. 0. 0.]
    #    [0. 0. 0. 0.]]]]