SpatialDropout2D 层

[源代码]

SpatialDropout2D 类

keras.layers.SpatialDropout2D(
    rate, data_format=None, seed=None, name=None, dtype=None
)

Spatial 2D Dropout 版本。

此版本执行与 Dropout 相同的功能，但它会丢弃整个 2D 特征图而不是单个元素。如果特征图中的相邻像素高度相关（通常在早期卷积层中是这种情况），那么常规 Dropout 将无法正则化激活，否则只会导致有效学习率下降。在这种情况下，SpatialDropout2D 将有助于促进特征图之间的独立性，应被使用。

参数

• rate：0 到 1 之间的浮点数。要丢弃的输入单元的比例。
• data_format："channels_first" 或 "channels_last"。在 "channels_first" 模式下，通道维度（深度）位于索引 1；在 "channels_last" 模式下，它位于索引 3。它默认为 Keras 配置文件 ~/.keras/keras.json 中的 image_data_format 值。如果您从未设置过，则默认为 "channels_last"。

调用参数

• inputs：一个 4D 张量。
• training：布尔值，指示层是应在训练模式下（应用 dropout）还是在推理模式下（直通）运行。

输入形状

4D 张量，形状为：(samples, channels, rows, cols) (如果 data_format='channels_first') 或 4D 张量，形状为：(samples, rows, cols, channels) (如果 data_format='channels_last')。

输出形状与输入相同。

参考

• Tompson 等人，2014 年