Keras – 重塑图层
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) 被设定为目标形状。