将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
转载请注明原文地址:https://ipadbbs.8miu.com/read-48910.html