sum instead of diag to compute std in std_fdatagrid integrand

eliegoudout · Oct 12, 2023 · 6b5e47d · 6b5e47d
1 parent 5509c00
commit 6b5e47d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/skfda/exploratory/stats/_stats.py b/skfda/exploratory/stats/_stats.py
@@ -146,7 +146,10 @@ def std_fdatabasis(X: FDataBasis, ddof: int = 1) -> FDataBasis:
     def std_function(t_points: NDArrayFloat) -> NDArrayFloat:  # noqa: WPS430
         basis_evaluation = basis(t_points).reshape((basis.n_basis, -1))
         std_values = np.sqrt(
-            np.diag(basis_evaluation.T @ coeff_cov_matrix @ basis_evaluation),
+            np.sum(
+                basis_evaluation * (coeff_cov_matrix @ basis_evaluation),
+                axis=0,
+            ),
         )
         return np.reshape(std_values, (1, -1, X.dim_codomain))