torch.nn.functional.gaussian_nll_loss¶
- torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean')[source][source]¶
高斯负对数似然损失。
有关详细信息，请参阅
GaussianNLLLoss。- 参数
input (Tensor) – 高斯分布的期望值。
target (Tensor) – 来自高斯分布的样本。
var (Union[Tensor, float]) – 正方差张量，输入中每个期望值对应一个（异方差），或者单个（同方差），或者用于所有期望值的正标量值。
full (bool, optional) – 在损失计算中包含常数项。默认值：
False。eps (float, optional) – 为稳定性添加到 var 的值。默认值：1e-6。
reduction (str, optional) – 指定应用于输出的 reduction 方式：
'none'|'mean'|'sum'。'none'：不应用 reduction，'mean'：输出是所有批次成员损失的平均值，'sum'：输出是所有批次成员损失的总和。默认值：'mean'。
- 返回类型
