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

Question

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

Answer 1

基于SciPy对马氏距离的实现，您可以在 PyTorch 中执行此操作。假设u和v是 1D 并且cov是 2D 协方差矩阵。

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采用协方差矩阵的逆矩阵。