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'。
- 返回型別
