Skip to content

Commit f511f52

Browse files
committed
Improved kerneldiagmatrix efficiency
1 parent 2bfd596 commit f511f52

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/matrix/kernelmatrix.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,9 +108,9 @@ function kerneldiagmatrix(
108108
)
109109
@assert obsdim [1,2] "obsdim should be 1 or 2 (see docs of kernelmatrix))"
110110
if obsdim == 1
111-
[@views _kernel(κ,X[i,:],X[i,:]) for i in 1:size(X,obsdim)]
111+
eachrow(X) .|> x->_kernel(κ,x,x)
112112
elseif obsdim == 2
113-
[@views _kernel(κ,X[:,i],X[:,i]) for i in 1:size(X,obsdim)]
113+
eachcol(X) .|> x->_kernel(κ,x,x)
114114
end
115115
end
116116

0 commit comments

Comments
 (0)