Просто возьмите функцию потерь из бэкэнда и используйте ее.
Пример - если включен режим ожидания:
losses = tf.keras.backend.categorical_crossentropy(true_data, pred_data)
Пример - если режим ожидания отключен:
def loss_calc(x):
return backend.categorical_crossentropy(x[0], x[1])
trueIn = Input(shape_of_the_targets)
predIn = Input(shape_of_the_targets)
out = Lambda(loss_calc)([trueIn, predIn])
loss_model = Model([trueIn, predIn], out)
losses = loss_model.predict([true_data, pred_data])