Pad1D¶
- class paddle.nn. Pad1D ( padding, mode='constant', value=0.0, data_format='NCL', name=None ) [源代码] ¶
Pad1D
按照 padding、mode 和 value 属性对输入进行填充。
- 参数:
-
padding (Tensor | List[int] | int) - 填充大小。如果是int,则在所有待填充边界使用相同的填充, 否则填充的格式为[pad_left, pad_right]。
mode (str) - padding的四种模式,分别为
'constant'
,'reflect'
,'replicate'
和'circular'
。'constant'
表示填充常数value
;'reflect'
表示填充以输入边界值为轴的映射;'replicate'
表示 填充输入边界值;'circular'
为循环填充输入。默认值为'constant'
。value (float32) - 以
'constant'
模式填充区域时填充的值。默认值为0.0。data_format (str) - 指定输入的format,可为
'NCL'
或者'NLC'
,默认值为'NCL'
。name (str, 可选) - 该参数供开发人员打印调试信息时使用,具体用法请参见 Name ,缺省值为None。
返回:无
代码示例
import paddle
import paddle.nn as nn
import numpy as np
input_shape = (1, 2, 3)
pad = [1, 2]
mode = "constant"
data = paddle.arange(np.prod(input_shape), dtype="float32").reshape(input_shape) + 1
my_pad = nn.Pad1D(padding=pad, mode=mode)
result = my_pad(data)
print(result)
# [[[0. 1. 2. 3. 0. 0.]
# [0. 4. 5. 6. 0. 0.]]]