KLDivLoss

    该算子计算输入(Input)和输入(Label)之间的Kullback-Leibler散度损失。注意其中输入(Input)应为对数概率值,输入(Label)应为概率值。

    kL发散损失计算如下:

    reductionmean 时,输出损失为[1]的形状,输出为所有损失的平均值。

    reduction 为 时,输出损失为[1]的形状,输出为所有损失的总和。

    reductionbatchmean 时,输出损失为[N]的形状,N为批大小,输出为所有损失的总和除以批量大小。

    形状:

    代码示例: