當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.compat.v1.layers.AveragePooling2D用法及代碼示例


2D 輸入(例如圖像)的平均池化層。

繼承自：AveragePooling2D、Layer、Layer、Module

用法

tf.compat.v1.layers.AveragePooling2D(
    pool_size, strides, padding='valid',
    data_format='channels_last', name=None, **kwargs
)

參數

  • pool_size 一個整數或 2 個整數的元組/列表：(pool_height, pool_width) 指定池化窗口的大小。可以是單個整數，為所有空間維度指定相同的值。
  • strides 一個整數或 2 個整數的元組/列表，指定池化操作的步長。可以是單個整數，為所有空間維度指定相同的值。
  • padding 一個字符串。填充方法，'valid' 或 'same'。不區分大小寫。
  • data_format 一個字符串。輸入中維度的排序。支持channels_last(默認)和channels_first。 channels_last 對應於形狀為 (batch, height, width, channels) 的輸入，而 channels_first 對應於形狀為 (batch, channels, height, width) 的輸入。
  • name 一個字符串，圖層的名稱。

屬性

  • graph
  • scope_name

遷移到 TF2

警告：這個 API 是為 TensorFlow v1 設計的。繼續閱讀有關如何從該 API 遷移到本機 TensorFlow v2 等效項的詳細信息。見TensorFlow v1 到 TensorFlow v2 遷移指南有關如何遷移其餘代碼的說明。

此 API 是一個遺留 api，僅與 Eager Execution 和 tf.function 兼容，如果您將其與 tf.compat.v1.keras.utils.track_tf1_style_variables 結合使用

請參閱遷移指南的 tf.layers 模型映射部分，了解如何在 TF2 中將 TensorFlow v1 模型與 Keras 一起使用。

對應的 TensorFlow v2 層是 tf.keras.layers.AveragePooling2D 。

到原生 TF2 的結構映射

支持的參數均未更改名稱。

前：

pooling = tf.compat.v1.layers.AveragePooling2D(pool_size=2, strides=2)

後：

pooling = tf.keras.layers.AveragePooling2D(pool_size=2, strides=2)

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.compat.v1.layers.AveragePooling2D。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。