Keras - Reshape 重塑图层
Reshape(重塑) 用于更改输入的形状。例如,如果将参数为 (2,3) 的重塑应用于输入形状为 (batch_size, 3, 2) 的图层,则该图层的输出形状将为 (batch_size, 2, 3)
Reshape 有一个参数,如下所示 −
keras.layers.v(target_shape)
使用 重塑图层 的一个简单示例如下 −
>>> from keras.models import Sequential >>> from keras.layers import Activation, Dense, Reshape >>> >>> >>> model = Sequential() >>> layer_1 = Dense(16, input_shape = (8,8)) >>> model.add(layer_1) >>> layer_2 = Reshape((16, 8)) >>> model.add(layer_2) >>> layer_2.input_shape (None, 8, 16) >>> layer_2.output_shape (None, 16, 8) >>>
其中,(16, 8)设置为目标形状。