tensorflow篇——lamda层

    技术2024-04-15  75

    将expand_dim, squeeze或者自定义层转换成keras中的Layer时,一般有两种方法,一种是直接定义类class然后继承Layer,一种是直接使用Lambda函数。 1.lamda层的构建方法如下:

    from tensorflow import keras squeeze = keras.layers.Lambda(lambda x: keras.backend.squeeze(x, axis=-1))(x
    Processed: 0.016, SQL: 9