torch.corrcoef¶
- torch.corrcoef(input) Tensor ¶
估计由
input
矩阵给定的变量的皮尔逊积矩相关系数矩阵,其中行表示变量,列表示观测值。注意
相关系数矩阵 R 使用协方差矩阵 C 计算,如下所示:
注意
由于浮点舍入,结果数组可能不是厄米特矩阵,其对角元素可能不为 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.)