Keras 2 API 文件 / 層 API / 形狀重塑層 / ZeroPadding1D 層

ZeroPadding1D 層

[原始碼]

ZeroPadding1D 類別

tf_keras.layers.ZeroPadding1D(padding=1, **kwargs)

用於 1D 輸入(例如時間序列)的零填充層。

範例

>>> input_shape = (2, 2, 3)
>>> x = np.arange(np.prod(input_shape)).reshape(input_shape)
>>> print(x)
[[[ 0  1  2]
  [ 3  4  5]]
 [[ 6  7  8]
  [ 9 10 11]]]
>>> y = tf.keras.layers.ZeroPadding1D(padding=2)(x)
>>> print(y)
tf.Tensor(
  [[[ 0  0  0]
    [ 0  0  0]
    [ 0  1  2]
    [ 3  4  5]
    [ 0  0  0]
    [ 0  0  0]]
   [[ 0  0  0]
    [ 0  0  0]
    [ 6  7  8]
    [ 9 10 11]
    [ 0  0  0]
    [ 0  0  0]]], shape=(2, 6, 3), dtype=int64)

引數

  • padding: 整數,或整數元組(長度為 2)。
    • 若為整數:在填充維度(軸 1)的開頭和結尾添加的零個數。
    • 若為整數元組(長度為 2):在填充維度的開頭和結尾添加的零個數 ((左邊填充, 右邊填充))。

輸入形狀

具有形狀 (batch_size, axis_to_pad, features) 的 3D 張量

輸出形狀

具有形狀 (batch_size, padded_axis, features) 的 3D 張量