Replace two-argument maps by broadcasts

JuliaGaussianProcesses · Oct 3, 2023 · 23a9638 · 23a9638
1 parent 8e3a4a3
commit 23a9638
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/space_time/pseudo_point.jl b/src/space_time/pseudo_point.jl
@@ -103,10 +103,11 @@ function kernel_diagonals(k::DTCSeparable, x::RegularInTime)
     space_kernel = k.k.l
     time_kernel = k.k.r
     time_vars = kernelmatrix_diag(time_kernel, get_times(x))
-    return map(
-        (s_t, x_r) -> Diagonal(kernelmatrix_diag(space_kernel, x_r) * s_t),
-        time_vars,
-        x.vs,
+    return Diagonal.(
+        kernelmatrix_diag.(
+            Ref(space_kernel),
+            x.vs
+        ) .* time_vars
     )
 end