快捷方式

torch.corrcoef

torch.corrcoef(input) Tensor

估计由 input 矩阵给定的变量的皮尔逊积矩相关系数矩阵,其中行表示变量,列表示观测值。

注意

相关系数矩阵 R 使用协方差矩阵 C 计算,如下所示:Rij=CijCiiCjjR_{ij} = \frac{ C_{ij} } { \sqrt{ C_{ii} * C_{jj} } }

注意

由于浮点舍入,结果数组可能不是厄米特矩阵,其对角元素可能不为 1。实部和虚部被剪辑到区间 [-1, 1],以尝试改善这种情况。

参数

input (Tensor) – 包含多个变量和观测值的二维矩阵,或表示单个变量的标量或一维向量。

返回值

(Tensor) 变量的相关系数矩阵。

另请参阅

torch.cov() 协方差矩阵。

示例

>>> x = torch.tensor([[0, 1, 2], [2, 1, 0]])
>>> torch.corrcoef(x)
tensor([[ 1., -1.],
        [-1.,  1.]])
>>> x = torch.randn(2, 4)
>>> x
tensor([[-0.2678, -0.0908, -0.3766,  0.2780],
        [-0.5812,  0.1535,  0.2387,  0.2350]])
>>> torch.corrcoef(x)
tensor([[1.0000, 0.3582],
        [0.3582, 1.0000]])
>>> torch.corrcoef(x[0])
tensor(1.)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获取答案

查看资源