Keras 2 API 文档 / 层 API / 正则化层 / SpatialDropout1D 层

SpatialDropout1D 层

[来源]

SpatialDropout1D

tf_keras.layers.SpatialDropout1D(rate, **kwargs)

Dropout 的空间 1D 版本。

此版本执行与 Dropout 相同的功能,但是它会丢弃整个 1D 特征图,而不是单个元素。如果特征图中相邻帧之间存在强相关性(在早期卷积层中通常如此),则常规 Dropout 将无法正则化激活,否则只会导致有效学习率下降。在这种情况下,SpatialDropout1D 将有助于促进特征图之间的独立性,因此应使用它。

参数

  • rate:0 到 1 之间的一个浮点数。要丢弃的输入单元的分数。

调用参数

  • inputs:一个 3D 张量。
  • training:Python 布尔值,指示层应该以训练模式(添加 Dropout)还是推理模式(不做任何事情)运行。

输入形状

形状为 (samples, timesteps, channels) 的 3D 张量

输出形状 与输入相同

.

参考文献: - 使用卷积神经网络进行高效的目标定位