Я нашел код, использующий 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]