from keras.layers import Lambda
lambda函数是一种匿名函数,即没有名字的函数,定义为 lambda 输入变量: 含输入的表达式
lambda x: K.mean(x, axis=-1, keepdims=True)
等效于
def channelmean(x):
x = K.mean(x, axis=-1, keepdims=True)
return x
out = Lambda(channelmean)(input)
Lambda层可以对输入数据进行任意数学表达式计算,并将计算结果输出:
a = Lambda(lambda x: K.mean(x, axis=1))(a)