torch.nn.functional.gaussian_nll_loss¶
- torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean')[源代码]¶
高斯负对数似然损失。
详情请参见
GaussianNLLLoss。- Parameters
输入 (张量) – 高斯分布的期望值。
目标 (张量) – 从高斯分布中采样的样本。
var (Tensor) – 正方差张量，每个输入中的期望值对应一个（异方差），或单个（同方差）。
full (bool, 可选) – 在损失计算中包含常数项。默认值：
False。eps (float, 可选) – 为了稳定性而添加到方差中的值。默认值：1e-6。
reduction (str, 可选) – 指定应用于输出的reduction方式：
'none'|'mean'|'sum'。'none'：不进行reduction，'mean'：输出是所有批次成员损失的平均值，'sum'：输出是所有批次成员损失的总和。 默认值：'mean'。
- Return type
