Keras定义模型有两种方法。

面向对象式的API

model = Sequential()
model.add(Dense(8, input_shape=(10,), activation="relu"))
model.add(Dense(4, activation="relu"))
model.add(Dense(1, activation="linear"))

函数式的API

inputs = Input(shape=(10,))
x = Dense(8, activation="relu")(inputs)
x = Dense(4, activation="relu")(x)
x = Dense(1, activation="linear")(x)
model = Model(inputs, x)

keras自定义计算需要用到lamda层

https://blog.csdn.net/weixin_43935696/article/details/112094223