单个指标返回多个值

Question

我最近从tensorflow v1 api升级到tensorflow v2 api，似乎被不良行为所困扰。我有一个指标，它返回一个在张量流 v1 中有效的数组。然而，keras 决定将我的指标合并为一个值，这是不受欢迎的行为。

我能做的就是调用相同的指标 6 次，但我假设它将独立计算 6 次，这是我真正想避免的。有没有办法让 keras 返回数组的所有 6 个值，即使是nx 6输出也比这种行为更好。

def iou_loss(self):
    def model_quality(p, y):
        # brevity

        return [true_positive, false_positive, false_negative, completeness, correctness, quality]

    return model_quality

我想在train和predict阶段分别计算这些指标。