PyTorch torch.cov 函数
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)
# 计算协方差矩阵
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 参考手册