python - pytorch 是否具有计算相关系数矩阵的功能，例如 numpy.corrcoef ()

Question

pytorch 是否具有计算相关系数矩阵的功能，例如numpy.corrcoef ()

标签： pythonnumpypytorch

Answer 1

不，它没有。但是如果你希望它比 numpy 更快，你可以在 pytorch 中自己实现它。这个线程可能会有所帮助。还有一个audtorch基于 pytorch 的库名称，似乎在这里实现了 pearson 相关系数。