torch.nn.functional.gaussian_nll_loss¶
- torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean')[源代码]¶
高斯负对数似然损失。
有关详细信息，请参见
GaussianNLLLoss。- 参数
input (张量) – 高斯分布的期望。
target (张量) – 来自高斯分布的样本。
var (张量) – 正方差张量，每个输入期望一个（异方差），或单个方差（同方差）。
full (布尔值, 可选) – 在损失计算中包含常数项。默认值：
False。eps (浮点数, 可选) – 添加到 var 的值，用于稳定性。默认值：1e-6。
reduction (字符串, 可选) – 指定要应用于输出的缩减：
'none'|'mean'|'sum'。'none'：不应用任何缩减，'mean'：输出是所有批次成员损失的平均值，'sum'：输出是所有批次成员损失的总和。默认值：'mean'。
- 返回类型
