Keras 3 API 文件 / 層 API / 正規化層 / SpatialDropout3D 層

SpatialDropout3D 層

[原始碼]

SpatialDropout3D 類別

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

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 張量,形狀為:(samples, channels, dim1, dim2, dim3),如果 data_format='channels_first';或 5D 張量,形狀為:(samples, dim1, dim2, dim3, channels),如果 data_format='channels_last'。

輸出形狀 與輸入相同。

參考文獻