Keras 2 API 文档 / 层API / 重塑层 / ZeroPadding1D 层

ZeroPadding1D 层

[source]

ZeroPadding1D class

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

Zero-padding layer for 1D input (e.g. temporal sequence).

Examples

>>> 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)

Arguments

  • padding: Int, or tuple of int (length 2).
    • If int: How many zeros to add at the beginning and end of the padding dimension (axis 1).
    • If tuple of int (length 2): How many zeros to add at the beginning and the end of the padding dimension ((left_pad, right_pad)).

Input shape

3D tensor with shape (batch_size, axis_to_pad, features)

Output shape

3D tensor with shape (batch_size, padded_axis, features)