Python tf.keras.layers.Reshape用法及代碼示例

將輸入重塑為給定形狀的層。

繼承自：Layer，Module

用法

tf.keras.layers.Reshape(
    target_shape, **kwargs
)

參數

target_shape 目標形狀。整數元組，不包括樣本維度(批量大小)。
**kwargs 任何其他層關鍵字參數。

輸入形狀：

任意的，盡管輸入形狀中的所有維度都必須是已知/固定的。將此層用作模型中的第一層時，請使用關鍵字參數input_shape(整數元組，不包括樣本/批量大小軸)。

輸出形狀：

(batch_size,) + target_shape

例子：

# as first layer in a Sequential model
model = tf.keras.Sequential()
model.add(tf.keras.layers.Reshape((3, 4), input_shape=(12,)))
# model.output_shape == (None, 3, 4), `None` is the batch size.
model.output_shape
(None, 3, 4)

# as intermediate layer in a Sequential model
model.add(tf.keras.layers.Reshape((6, 2)))
model.output_shape
(None, 6, 2)

# also supports shape inference using `-1` as dimension
model.add(tf.keras.layers.Reshape((-1, 2, 2)))
model.output_shape
(None, 3, 2, 2)

相關用法

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