torch.nn.functional.gaussian_nll_loss¶
- torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean')[source][source]¶
Gaussian 負對數概似損失函數。
詳情請參閱
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'。
- 返回類型
