现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.cov 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

torch.cov 是 PyTorch 中用于计算协方差矩阵的函数。它计算输入张量的协方差矩阵,用于衡量变量之间的协方差。

函数定义

torch.cov(input, correction=1, fweights=None, aweights=None)

使用示例

实例

import torch

# 计算协方差矩阵
x = torch.tensor([[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6]], dtype=torch.float32)
print("输入矩阵:")
print(x)

# 计算协方差矩阵(默认使用贝塞尔校正)
cov = torch.cov(x)
print("协方差矩阵:")
print(cov)

# 不使用校正 (correction=0)
cov_no_correction = torch.cov(x, correction=0)
print("无校正协方差矩阵:")
print(cov_no_correction)

# 两个变量
a = torch.tensor([1, 2, 3, 4, 5], dtype=torch.float32)
b = torch.tensor([2, 4, 6, 8, 10], dtype=torch.float32)
data = torch.stack([a, b])
cov2 = torch.cov(data)
print("a 和 b 的协方差矩阵:")
print(cov2)

Pytorch torch 参考手册 Pytorch torch 参考手册