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

展平輸入張量，同時保留批處理軸(軸 0)。

用法

tf.compat.v1.layers.flatten(
    inputs, name=None, data_format='channels_last'
)

參數

inputs 張量輸入。
name 圖層的名稱(字符串)。
data_format 一個字符串，是 channels_last (默認)或 channels_first 之一。輸入中維度的排序。 channels_last 對應於形狀為 (batch, height, width, channels) 的輸入，而 channels_first 對應於形狀為 (batch, channels, height, width) 的輸入。

重塑張量。

遷移到 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.Flatten 。

到原生 TF2 的結構映射

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

前：

y = tf.compat.v1.layers.flatten(x)

後：

要使用 TF1 函數層遷移代碼，請使用 Keras 函數 API：

x = tf.keras.Input((28, 28, 1))
 y = tf.keras.layers.Flatten()(x)
 model = tf.keras.Model(x, y)

例子：

x = tf.compat.v1.placeholder(shape=(None, 4, 4), dtype='float32')
  y = flatten(x)
  # now `y` has shape `(None, 16)`

  x = tf.compat.v1.placeholder(shape=(None, 3, None), dtype='float32')
  y = flatten(x)
  # now `y` has shape `(None, None)`

相關用法

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

用法

參數

返回

遷移到 TF2

到原生 TF2 的結構映射

例子：