當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.keras.layers.ZeroPadding1D用法及代碼示例


一維輸入的零填充層(例如時間序列)。

繼承自:LayerModule

用法

tf.keras.layers.ZeroPadding1D(
    padding=1, **kwargs
)

參數

  • padding Int,或 int 的元組(長度為 2),或字典。
    • 如果是 int:在填充維度(軸 1)的開頭和結尾添加多少個零。
    • 如果 int 元組(長度 2):在填充維度的開頭和結尾添加多少個零((left_pad, right_pad))。

例子:

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)

輸入形狀:

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

輸出形狀:

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

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.layers.ZeroPadding1D。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。