什么是keras.backend.tf?

时间:2019-06-04 02:28:41

标签: keras

我使用keras.backend.tf找到了一些代码。但是我找不到有关此库的任何信息。我​​想知道keras.backend.tf是否等于tensorflow,以便我可以使用keras.backend.tf在tf中使用任何函数。例如,就像tf.argmx一样,我可以使用keras.backend.tf.argmax。

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]

0 个答案:

没有答案