zeropad2d¶
返回一个按照 padding
属性对 x
进行零填充的 Tensor,数据类型与 x
相同。
参数¶
x (Tensor) - Tensor,format 可以为
'NCHW'
,'NHWC'
,默认值为'NCHW'
,数据类型支持 float16, float32, float64, int32, int64。padding (Tensor | List[int] | Tuple[int]) - 填充大小。pad 的格式为[pad_left, pad_right, pad_top, pad_bottom];
data_format (str,可选) - 指定
x
的 format,可为'NCHW'
,'NHWC'
,默认值为'NCHW'
。name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
Tensor:对
x
进行'pad'
的结果,数据类型和x
相同。
代码示例¶
import paddle
import paddle.nn.functional as F
x_shape = paddle.to_tensor([1, 1, 2, 3])
x = paddle.arange(paddle.prod(x_shape), dtype="float32").reshape(x_shape) + 1
y = F.zeropad2d(x, [1, 2, 1, 1])
print(y)
# [[[[0. 0. 0. 0. 0. 0.]
# [0. 1. 2. 3. 0. 0.]
# [0. 4. 5. 6. 0. 0.]
# [0. 0. 0. 0. 0. 0.]]]]