如何在pytorch中计算马氏距离？

【问题标题】：how to calculate mahalanobis distance in pytorch?如何在pytorch中计算马氏距离？
【发布时间】：2020-12-16 18:17:12
【问题描述】：

计算马氏距离最有效的方法是什么：在pytorch中？

【问题讨论】：

【解决方案1】：

基于马氏距离的SciPy's implementation，您可以在 PyTorch 中执行此操作。假设u 和v 是一维的，cov 是二维协方差矩阵。

def mahalanobis(u, v, cov):
    delta = u - v
    m = torch.dot(delta, torch.matmul(torch.inverse(cov), delta))
    return torch.sqrt(m)

注意：scipy.spatial.distance.mahalanobis 接受协方差矩阵的逆。

【讨论】：