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

SpatialDropout3D 层

[来源]

SpatialDropout3D

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

Spatial Dropout 的 3D 版本。

此版本执行与 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)还是推理模式(直通)下运行。

输入形状

5D 张量,形状:如果 data_format='channels_first',则为 (samples, channels, dim1, dim2, dim3);如果 data_format='channels_last',则为 5D 张量,形状:(samples, dim1, dim2, dim3, channels)

输出形状 与输入相同。

参考