Kerasの自作損失関数を実装したい

Question

前提・実現したいこと

Kerasで自作損失関数を作ろうとしています。

目的変数3つ(y[0],y[1],y[2])に対し

{(y_pred[0] - y_true[0])^2+(y_pred[0] - y_true[0] + y_pred[1] - y_true[1])^2}/2

となるような関数が作りたいです。

発生している問題・エラーメッセージ

目的変数をそれぞれ区別して計算する必要があり、K.meanといったkerasのbackend関数が使えず実装することができませんでした。

試したこと

def mean_squared_error(y_true, y_pred):
    return K.mean(K.square(y_pred - y_true), axis=-1)

KerasのMSEの実装を参考にしながらテンソル型について色々調べながら実装を試みましたがうまくいきませんでした。
損失関数の引数であるy_true,y_predを出力したところ、shapeが(?, ?)となっていました。
y_trueとy_predには実行された時点では値が入ってないと推測しましたが、この場合どうやって実装すればいいのでしょうか。