cov_inv = torch.linalg.inv(cov)返回nan

如果 `torch.linalg.inv(cov)` 返回 `nan`，则可能是因为协方差矩阵 `cov` 是奇异的（singular），也就是说，它的行列式为0，无法求逆。这种情况通常出现在数据集中的某些特征之间存在线性相关性的情况下。

解决这个问题的一种方法是使用伪逆（pseudo-inverse）来代替逆矩阵。可以使用 `torch.pinverse()` 函数来计算伪逆矩阵，例如：

cov_inv = torch.pinverse(cov)

`torch.pinverse()` 函数能够处理奇异矩阵，但是它计算的是伪逆矩阵，不同于普通的逆矩阵，因此在使用时需要注意。

另外，如果你的协方差矩阵非常大，可能需要使用更高效的矩阵求逆方法，例如使用基于LU分解的方法，可以使用 `torch.lu()` 函数来实现。