如何在tf.keras自定义损失函数内部触发python函数？

import tensorflow as tf

def custom_loss(y_true, y_pred):
    # 在自定义损失函数内部触发Python函数
    result = my_python_function(y_true, y_pred)
    loss = tf.reduce_mean(result)
    return loss

def my_python_function(y_true, y_pred):
    # 在这里编写你的Python函数逻辑
    # 可以对y_true和y_pred进行任何操作
    result = tf.square(y_true - y_pred)
    return result

# 使用自定义损失函数
model.compile(optimizer='adam', loss=custom_loss)