TimeDistributed 层

[源代码]

TimeDistributed 类

tf_keras.layers.TimeDistributed(layer, **kwargs)

此包装器允许将一个层应用于输入的每个时间切片。

每个输入至少应为 3D，第一个输入的索引 1 的维度将被视为时间维度。

考虑一个包含 32 个视频样本的批次，其中每个样本是 128x128 的 RGB 图像，使用 channels_last 数据格式，跨 10 个时间步长。批次输入形状为 (32, 10, 128, 128, 3)。

然后，您可以使用 TimeDistributed 将相同的 Conv2D 层独立地应用于每个 10 个时间步长。

>>> inputs = tf.keras.Input(shape=(10, 128, 128, 3))
>>> conv_2d_layer = tf.keras.layers.Conv2D(64, (3, 3))
>>> outputs = tf.keras.layers.TimeDistributed(conv_2d_layer)(inputs)
>>> outputs.shape
TensorShape([None, 10, 126, 126, 64])

因为 TimeDistributed 将 Conv2D 的同一实例应用于每个时间戳，所以在每个时间戳都使用相同的权重集。

参数

• layer：一个 tf.keras.layers.Layer 实例。

调用参数

• inputs：形状为 (batch, time, ...) 或嵌套张量的输入张量，并且每个张量的形状都为 (batch, time, ...)。
• training：Python 布尔值，指示层应在训练模式下还是推理模式下运行。此参数传递给包装的层（仅当层支持此参数时）。
• mask：形状为 (samples, timesteps) 的二进制张量，指示是否应该屏蔽给定的时间步长。此参数传递给包装的层（仅当层支持此参数时）。

引发异常

• ValueError：如果未初始化为 tf.keras.layers.Layer 实例。