torch.nn.functional.gaussian_nll_loss#
- torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean')[source]#
计算高斯负对数似然损失。
有关详细信息，请参阅
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'。
- 返回类型