Python PyTorch corrcoef用法及代碼示例

本文簡要介紹python語言中 torch.corrcoef 的用法。

用法:

torch.corrcoef(input) → Tensor

參數：

input(Tensor) -包含多個變量和觀察值的 2D 矩陣，或表示單個變量的標量或 1D 向量。

返回：

(張量)變量的相關係數矩陣。

估計 input 矩陣給出的變量的 Pearson product-moment 相關係數矩陣，其中行是變量，列是觀察值。

注意

相關係數矩陣 R 使用協方差矩陣 C 計算，如下所示R_{ij} = \frac{ C_{ij} } { \sqrt{ C_{ii} * C_{jj} } }

注意

由於浮點舍入，結果數組可能不是 Hermitian，其對角線元素可能不是 1。實數和虛數被裁剪到區間 [-1, 1] 以試圖改善這種情況。

例子：

>>> 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.)