我使用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]