损失函数中的tf.boolean_mask：没有为任何变量提供渐变

Question

我正在尝试使用tf.boolean_mask获得用于图像分割的被掩盖均值差异：

def custom_loss(image):
    def loss(predicted_y, target_y):
        pred_mask = tf.math.greater(predicted_y,0.5)
        target_mask = tf.math.greater(target_y,0.5)
        mean_diff = (tf.reduce_mean(tf.boolean_mask(image,pred_mask)) - tf.reduce_mean(tf.boolean_mask(image,target_mask))) ** 2
        return mean_diff
return loss

不幸的是，我遇到了ValueError：没有为任何变量提供渐变，这在逻辑上是由tf.boolean_mask引起的。在Tensorflow 2.0上有任何方法吗？

非常感谢！