Keras 3 API 文档 / 层 API / 正则化层 / SpatialDropout2D 层

SpatialDropout2D 层

[源代码]

SpatialDropout2D

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

Dropout 的空间 2D 版本。

此版本执行与 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 张量,形状为:如果 data_format='channels_first',则为 (样本, 通道, 行, 列);如果 data_format='channels_last',则为 (样本, 行, 列, 通道)

输出形状与输入相同。

参考