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) – 添加到方差中的值，用于稳定性。默认值: 1e-6。
reduction (str, optional) – 指定应用于输出的归约方式:
'none'|'mean'|'sum'。'none': 不应用归约，'mean': 输出是所有批处理成员损失的平均值，'sum': 输出是所有批处理成员损失的总和。默认值:'mean'。
- 返回类型
