параметры перехвата в пользовательской функции потерь в Keras

#keras

Вопрос:

Я пытаюсь определить свою пользовательскую функцию потерь, которая является многомасштабной SSIM в моей модели keras. Однако я не знаю, как получить форму y_pred и y_true, потому что в процессе обучения они оба являются объектом «Нет».

Например.

 def custom_loss(y_pred, y_true):
    print(y_pred.get_shape())
    ...
    ...
    return something
 

появится сообщение об ошибке y_pred — Нет. Может кто-нибудь помочь, пожалуйста.