Reshape
类keras.layers.Reshape(target_shape, **kwargs)
将输入重塑为给定形状的层。
参数
输入形状
任意,但输入形状中的所有维度必须已知/固定。当将此层用作模型中的第一层时,请使用关键字参数 input_shape
(整数元组,不包括样本/批次大小轴)。
输出形状
(batch_size, *target_shape)
示例
>>> x = keras.Input(shape=(12,))
>>> y = keras.layers.Reshape((3, 4))(x)
>>> y.shape
(None, 3, 4)
>>> # also supports shape inference using `-1` as dimension
>>> y = keras.layers.Reshape((-1, 2, 2))(x)
>>> y.shape
(None, 3, 2, 2)