如何显示图像传递给损失函数？

Question

我有一个骰子丢失功能，可以正常工作。但是我想显示传递给该函数的图像，然后再计算骰子值。 （提前感谢）

例如： y_pred图像处于张量。我不知道如何在数组中显示它。

def dice_loss(y_true,y_pred)

    #i want to see how the y_pred image looks like here

    y_true_f = K.flatten(y_true)
    y_pred_f = K.flatten(y_pred)
    intersection = K.sum(y_true_f * y_pred_f)
    answer = (2. * intersection + smooth) / (K.sum(y_true_f) + 
    K.sum(y_pred_f) + smooth)
    return -answer