Что такое keras.backend.tf? - PullRequest
0 голосов
/ 04 июня 2019

Я нашел код, использующий keras.backend.tf.Но я не смог найти никакой информации об этой библиотеке. Интересно, равен ли keras.backend.tf тензору потока, чтобы я мог использовать любую функцию в tf keras.backend.tf.Например, я могу использовать keras.backend.tf.argmax так же, как tf.argmx.

from keras.engine import Layer
import keras.backend as K
    def call(self, inputs, **kwargs):
        padding = self.padding
        pool_size = self.pool_size
        strides = self.strides
        if K.backend() == 'tensorflow':
            ksize = [1, pool_size[0], pool_size[1], 1]
            padding = padding.upper()
            strides = [1, strides[0], strides[1], 1]
            output, argmax = K.tf.nn.max_pool_with_argmax(
                inputs,
                ksize=ksize,
                strides=strides,
                padding=padding)
        else:
            errmsg = '{} backend is not supported for layer {}'.format(
                K.backend(), type(self).__name__)
            raise NotImplementedError(errmsg)
        argmax = K.cast(argmax, K.floatx())
        return [output, argmax]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...