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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。