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

SpatialDropout2D 层

[source]

SpatialDropout2D class

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

空间二维版本的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: Python布尔值,指示该层是否应在训练模式(应用dropout) 或推理模式(直通)下运行.

输入形状: 如果data_format='channels_first',则为形状为(samples, channels, rows, cols)的4D张量 或者如果data_format='channels_last',则为形状为(samples, rows, cols, channels)的4D张量.

输出形状: 与输入相同.

参考: