def dice_tf(label, pred): TP = tf.reduce_sum(tf.mul(pred, label)) FP = tf.reduce_sum(tf.mul(pred, 1-label)) FN = tf.reduce_sum(tf.mul(1-pred, label)) return tf.truediv(2*TP, FP+FN+2*TP)