zeropad2d¶

paddle.nn.functional. zeropad2d ( x, padding, data_format='NCHW', name=None ) [源代码] ¶

返回一个按照 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)
Tensor(shape=[1, 1, 4, 6], dtype=float32, place=Place(cpu), stop_gradient=True,
[[[[0., 0., 0., 0., 0., 0.],
   [0., 1., 2., 3., 0., 0.],
   [0., 4., 5., 6., 0., 0.],
   [0., 0., 0., 0., 0., 0.]]]])

         