快捷鍵

torchaudio.functional.frechet_distance

torchaudio.functional.frechet_distance(mu_x, sigma_x, mu_y, sigma_y)[來源]

計算兩個多變量常態分佈之間的 Fréchet 距離 [Dowson and Landau, 1982]

具體來說,對於多變量高斯分佈 \(X(\mu_X, \Sigma_X)\)\(Y(\mu_Y, \Sigma_Y)\),此函數計算並傳回 \(F\),如下所示

\[F(X, Y) = || \mu_X - \mu_Y ||_2^2 + \text{Tr}\left( \Sigma_X + \Sigma_Y - 2 \sqrt{\Sigma_X \Sigma_Y} \right) \]
參數:
  • mu_x (torch.Tensor) – 多變量高斯分佈 \(\mu_X\) 的平均值,形狀為 (N,)

  • sigma_x (torch.Tensor) – \(X\) 的共變異數矩陣 \(\Sigma_X\),形狀為 (N, N)

  • mu_y (torch.Tensor) – 多變量高斯分佈 \(\mu_Y\) 的平均值,形狀為 (N,)

  • sigma_y (torch.Tensor) – \(Y\) 的共變異數矩陣 \(\Sigma_Y\),形狀為 (N, N)

傳回:

\(X\)\(Y\) 之間的 Fréchet 距離。

傳回類型:

torch.Tensor

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得適合初學者和進階開發人員的深入教學課程

檢視教學課程

資源

尋找開發資源並獲得問題解答

檢視資源