ZeroPad1D

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

ZeroPad1D

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

参数

  • padding (Tensor | List[int] | int) - 如果输入数据类型为 int,则在所有待填充边界使用相同的填充, 否则填充的格式为[pad_left, pad_right]。

  • data_format (str) - 指定输入的 format,可为 'NCL' 或者 'NLC',默认值为 'NCL',其中 N 是 batch size, C 是通道数, L 输入特征的长度。

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

返回

形状

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

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

代码示例

>>> import paddle
>>> import paddle.nn as nn

>>> input_shape = (1, 2, 3)
>>> pad = [1, 2]
>>> data = paddle.arange(paddle.prod(paddle.to_tensor(input_shape)), dtype="float32").reshape(input_shape) + 1
>>> my_pad = nn.ZeroPad1D(padding=pad)
>>> result = my_pad(data)
>>> print(result)
Tensor(shape=[1, 2, 6], dtype=float32, place=Place(cpu), stop_gradient=True,
[[[0., 1., 2., 3., 0., 0.],
  [0., 4., 5., 6., 0., 0.]]])