torch.nn.functional.gaussian_nll_loss
- torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean')[源代码]
-
高斯负对数似然损失函数。
详情请参阅
GaussianNLLLoss。- 参数
-
-
输入 (Tensor) – 表示高斯分布期望值的张量。
-
target (Tensor) – 从高斯分布中抽取的样本。
-
var (Tensor) – 输入中的每个期望对应一个正方差张量（异质），或所有期望共用一个方差（同质）。
-
full (bool, 可选) – 是否在损失计算中包含常数项。默认值：
False。 -
eps (float, 可选) – 用于稳定性而添加到 var 的值。默认值：1e-6。
-
reduction (str, 可选) – 指定要应用于输出的缩减方式：
'none'|'mean'|'sum'。'none': 不进行任何缩减，'mean': 输出是所有批次成员损失值的平均值，'sum': 输出是所有批次成员损失值的总和。默认：'mean'。
-
- 返回类型