torch.nn.functional.gaussian_nll_loss#
- torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean')[源码]#
计算高斯负对数似然损失。
有关详细信息，请参阅
GaussianNLLLoss。- 参数
input (Tensor) – 高斯分布的期望。
target (Tensor) – 高斯分布的样本。
var (Union[Tensor, float]) – 方差（们）的张量，对于输入中的每个期望有一个（异方差），或者一个单一的方差（同方差），或者一个正的标量值用于所有期望。
full (bool, optional) – 是否在损失计算中包含常数项。默认值：
False。eps (float, optional) – 添加到 var 中的值，用于提高稳定性。默认值：1e-6。
reduction (str, optional) – 指定应用于输出的约简方式：
'none'|'mean'|'sum'。'none'：不应用约简，'mean'：输出是所有批次成员损失的平均值，'sum'：输出是所有批次成员损失的总和。默认值：'mean'。
- 返回类型
