class LossManager: def __init__(self): pass def compute_loss(self, outputs): """ 计算流型损失的逻辑 :param outputs: 模型的输出 :return: 计算得到的流型损失值 """ # 计算流型损失(这里使用均方误差作为示例) manifold_loss = (outputs.abs()).mean() # 计算流型损失 return manifold_loss