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) – 新增到 var 中的值，用於提高穩定性。預設值：1e-6。
reduction (str, optional) – 指定應用於輸出的約簡方式：
'none'|'mean'|'sum'。'none'：不應用約簡，'mean'：輸出是所有批次成員損失的平均值，'sum'：輸出是所有批次成員損失的總和。預設值：'mean'。
- 返回型別
