[错误] 在 beta=0.5 时, Megatron GKD 中的 JSD 损失为负数; 同样的输入, FP32 则给出了正数结果