Keras 3 API 文档 / 层 API / 正则化层 / 空间丢弃3D层

空间丢弃3D层

[source]

SpatialDropout3D class

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

空间3D版本的Dropout.

此版本执行与Dropout相同的功能,然而,它丢弃整个3D特征图而不是单个元素.如果特征图中的相邻体素强烈相关(通常在早期的卷积层中是这种情况),那么常规的Dropout将不会正则化激活,否则只会导致有效的学习率下降.在这种情况下,SpatialDropout3D将有助于促进特征图之间的独立性,应该使用它来代替.

参数: rate: 介于0和1之间的浮点数.要丢弃的输入单元的比例. data_format: "channels_first""channels_last". 在"channels_first"模式下,通道维度(深度)在索引1处,在"channels_last"模式下在索引4处. 它默认为在您的Keras配置文件~/.keras/keras.json中找到的image_data_format值. 如果您从未设置过它,那么它将是"channels_last".

调用参数: inputs: 一个5D张量. training: Python布尔值,指示该层是否应在训练模式(应用dropout)或推理模式(直通)下运行.

输入形状: 如果data_format='channels_first',则为形状为(samples, channels, dim1, dim2, dim3)的5D张量 或者如果data_format='channels_last',则为形状为(samples, dim1, dim2, dim3, channels)的5D张量.

输出形状: 与输入相同.

参考: