当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。