ZeroPad1D¶
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.]]])