Pad2D

class paddle.nn.Pad2D ( padding, mode=’constant’, value=0.0, data_format=’NCHW’, name=None ) [源代码]

Pad2D

按照 padding、mode 和 value 属性对输入进行填充。

参数：

  • padding (Tensor | List[int32]) - 填充大小。pad的格式为[pad_left, pad_right, pad_top, pad_bottom]。

  • mode (str) - padding的四种模式，分别为 ‘constant’, ‘reflect’, ‘replicate’ 和`‘circular’。 `‘constant’ 表示填充常数 value；’reflect’ 表示填充以input边界值为轴的映射；’replicate’ 表示 填充input边界值；’circular’`为循环填充input。默认值为 `‘constant’ 。

  • value (float32) - 以 ‘constant’ 模式填充区域时填充的值。默认值为0.0。

  • data_format (str) - 指定input的format，可为 ‘NCHW’ 或者 ‘NHWC’，默认值为`‘NCHW’`。

  • name (str, 可选) - 该参数供开发人员打印调试信息时使用，具体用法请参见 Name ，缺省值为None。

返回：无

代码示例

  1. import paddle
  2. import paddle.nn as nn
  3. import numpy as np
  4. input_shape = (1, 1, 2, 3)
  5. pad = [1, 0, 1, 2]
  6. mode = "constant"
  7. data = paddle.arange(np.prod(input_shape), dtype="float32").reshape(input_shape) + 1
  8. my_pad = nn.Pad2D(padding=pad, mode=mode)
  9. result = my_pad(data)
  10. print(result)
  11. # [[[[0. 0. 0. 0.]
  12. #    [0. 1. 2. 3.]
  13. #    [0. 4. 5. 6.]
  14. #    [0. 0. 0. 0.]
  15. #    [0. 0. 0. 0.]]]]