[keras] 텐서플로 함수 케라스에서 사용하기
tensorflow layer to keras layer
텐서 플로에서 사용하던 함수나 레이어를 케라스에서 제공하지 않는 경우가 있다. 이 경우 텐서플로에서 제공하는 기능을 그대로 케라스에서 쉽게 사용할 수 있는 방법이 있다.
Keras Lambda
from keras.layers import Lambda
import keras.backend as K
# prev_layer : keras의 layer로 l2_norm layer의 이전 레이어.
keras_l2_norm = Lambda(lambda data : K.l2_normalize(x=data))(prev_layer)
tensorflow에서 제공하는 l2_normalize 함수를 keras layer로 만드는 방법이다. keras layer는 이전 레이어의 정보를 담고있어서 Lambda 함수를 사용해서 keras 객체로 만들어준뒤에 이전 레이어 정보인 prev_layer
를 입력으로 주어야 한다.
댓글
댓글 쓰기