Función torch.cov de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.covEs una función de PyTorch utilizada para calcular la matriz de covarianza. Calcula la matriz de covarianza del tensor de entrada, utilizada para medir la covarianza entre variables.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Calcular la matriz de covarianza
x = torch.tensor([[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6]], dtype=torch.float32)
print("Matriz de entrada:")
print(x)

# Calcular la matriz de covarianza (usando la corrección de Bessel por defecto)
cov = torch.cov(x)
print("Matriz de covarianza:")
print(cov)

# Sin corrección (correction=0)
cov_no_correction = torch.cov(x, correction=0)
print("Matriz de covarianza sin corrección:")
print(cov_no_correction)

# Dos variables
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("Matriz de covarianza de a y b:")
print(cov2)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones