Keras /张量流：对张量中的特定尺寸求和

Question

我想通过以下方式修改此骰子丢失功能：

from keras import backend as K
def dice_coef(y_true, y_pred):    
    y_true_f = K.flatten(y_true)
    y_pred_f = K.flatten(y_pred)
    intersection = K.sum(y_true_f * y_pred_f)
    return ((2. * intersection + smooth) / (K.sum(y_true_f) + K.sum(y_pred_f) + smooth))

y_pred的尺寸为batch_size, img_height, img_width, num_classes，我想为每个像素而不是每个类计算dice_coef。因此，对于每个像素，尺寸为1,1,1,num_classes。而且我希望能够首先使用类的不同值为每个像素计算骰子，以产生batch_size, img_height, img_width的尺寸。

但是，我不能只修改那个频道。