Эквивалент math_ops.floor в Керасе - PullRequest
       33

Эквивалент math_ops.floor в Керасе

0 голосов
/ 05 сентября 2018

Я пытаюсь реализовать пользовательский слой в Keras, где мне нужно преобразовать тензор с плавающей точкой [a, 1+a) в двоичный тензор для маскировки. Я вижу, что у Tensorflow есть функция floor, которая может это делать, но у Keras, похоже, ее нет в keras.backend. Есть идеи, как мне это сделать?

1 Ответ

0 голосов
/ 05 сентября 2018

По просьбе ОП, я упомяну ответ, который я дал в своем комментарии, и уточню подробнее:

Краткий ответ: вы не столкнетесь с какими-либо серьезными проблемами, если будете использовать tf.floor().

Длинный ответ: Использование внутренних функций Keras (т. Е. keras.backend.*) необходимо в тех случаях, когда 1) необходимо предварительно обработать или дополнить аргумент (ы), передаваемый фактической функции Tensorflow или Theano backend или постобработка возвращенных результатов. Например, метод mean в бэкэнде также может работать с булевыми тензорами в качестве входных данных, однако метод reduce_mean в TF ожидает числовые типы в качестве входных данных; или 2) вы хотите написать модель, которая будет работать со всеми поддерживаемыми Keras бэкэндами.

В противном случае, можно использовать большинство реальных бэкэнд-функций напрямую; однако, если функция была определена в модуле keras.backend, рекомендуется использовать ее вместо этого.

...