Fix the cov function

This commit is contained in:
Vishal Patil
2024-09-02 14:16:14 -04:00
committed by GitHub
parent 9dca0a8a0a
commit 288f7f2c70

View File

@@ -39,7 +39,7 @@ fn cov(data: &Tensor, device: &Device) -> Result<Tensor> {
let cov = centered let cov = centered
.transpose(D::Minus1, D::Minus2)? .transpose(D::Minus1, D::Minus2)?
.matmul(&centered)? .matmul(&centered)?
.broadcast_div(&Tensor::new(m as f32, device)?)?; .broadcast_div(&Tensor::new((m - 1) as f32, device)?)?;
Ok(cov) Ok(cov)
} }