ZeroPad2D

class paddle.nn. ZeroPad2D ( padding, data_format='NCHW', name=None ) [源代码]

ZeroPad2D

按照 padding 属性对输入进行零填充。

参数

  • padding (Tensor | List[int] | int]) - 填充大小。如果是 int,则在所有待填充边界使用相同的填充, 否则填充的格式为[pad_left, pad_right, pad_top, pad_bottom]。

  • data_format (str) - 指定输入的 format,可为 'NCHW' 或者 'NHWC',默认值为 'NCHW'

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

形状

  • x(Tensor): ZeroPadD 层的输入,要求形状为 4-D,dtype 为 'float32''float64'

  • output(Tensor):输出,形状为 4-D,dtype 与 'input' 相同

代码示例

import paddle
import paddle.nn as nn
import numpy as np

input_shape = (1, 1, 2, 3)
pad = [1, 0, 1, 2]
data = paddle.arange(np.prod(input_shape), dtype="float32").reshape(input_shape) + 1

my_pad = nn.ZeroPad2D(padding=pad)
result = my_pad(data)

print(result)
# [[[[0. 0. 0. 0.]
#    [0. 1. 2. 3.]
#    [0. 4. 5. 6.]
#    [0. 0. 0. 0.]
#    [0. 0. 0. 0.]]]]