1、Lambda 层
如果你只是想对流经该层的数据做个变换,而这个变换本身没有什么需要学习的参数,那么直接用Lambda Layer是最合适的了。
- <span style="font-size:18px;">def get_submean_model():
- model = Sequential()
- model.add(Dense(5,input_dim=7))
- def sub_mean(x):
- x -= K.mean(x,axis=1,keepdims=True)
- return x
- model.add( Lambda(sub_mean,output_shape=lambda input_shape:input_shape))
- model.compile(optimizer='rmsprop',loss='mse')
- return model
- model = get_submean_model()
- res=model.predict(np.random.random((3,7)))</span>