ZeroPad2D¶
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.]]]]