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。- 参数:
输入（张量）- 高斯分布的期望。
目标（张量）- 从高斯分布中抽取的样本。
var（Union[Tensor, float]）- 正方差（正方差）的张量，每个期望一个，或单个值（同方差），或用于所有期望的正标量值。
full（bool，可选）- 在损失计算中包含常数项。默认：
False。eps（浮点数，可选）- 添加到 var 的值，用于稳定性。默认：1e-6。
reduction（字符串，可选）- 指定应用于输出的降维方式：
'none'|'mean'|'sum'.'none': 不应用降维，'mean': 输出为所有批次成员损失的均值，'sum': 输出为所有批次成员损失的总和。默认：'mean'。
- 返回类型:
