keras 使用Lambda 快速新建层 添加多个参数

keras许多简单操作,都需要新建一个层,使用Lambda可以很好完成需求。

# 额外参数
def normal_reshape(x, shape):
    return K.reshape(x,shape)

output = Lambda(normal_reshape, arguments={'shape':(-1, image_seq, 1000)})(output)
output = Lambda(lambda inp: K.mean(inp, axis=1), output_shape=(1000,))(output)

更多参考

猜你喜欢

转载自blog.csdn.net/qq_16234613/article/details/82724981
今日推荐