Есть ли способ сравнить в слоях keras?

Я хочу сравнить два числа в модели keras. Входом этого слоя является тензорная переменная, и этот слой сравнивает эту тензорную переменную с константой. Тогда он вернет 0 или 1. Есть ли какой-нибудь метод? Я попытался найти в анано функцию для этой работы, но не смог.


keras theano
person Haoran Wu    schedule 15.09.2017    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Вы можете найти функции в бэкэнде keras

import keras.backend as K

Вам понадобится одно из следующих: K.equal, K.greater, K.greater_equal и т. Д.

Для этого можно использовать слой Lambda:

Lambda(lambda x: K.cast(K.greater_equal(x,constant),'float32'),output_shape=sameAsInputShape)
person Daniel Möller    schedule 15.09.2017