Keras 3 API 文档 / 层 API / 重塑层 / 重塑层

重塑层

[source]

Reshape class

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

将输入重塑为给定形状的层.

参数: target_shape: 目标形状.整数的元组,不包括样本维度(批量大小).

输入形状: 任意,尽管输入形状中的所有维度必须是已知/固定的.当将此层用作模型中的第一层时,请使用关键字参数 input_shape(整数的元组,不包括样本/批量大小轴).

输出形状: (batch_size, *target_shape)

示例:

>>> x = keras.Input(shape=(12,))
>>> y = keras.layers.Reshape((3, 4))(x)
>>> y.shape
(None, 3, 4)
>>> # 也支持使用 `-1` 作为维度进行形状推断
>>> y = keras.layers.Reshape((-1, 2, 2))(x)
>>> y.shape
(None, 3, 2, 2)