@@ -101,7 +101,8 @@ function kernelmatrix(
101
101
_kernelmatrix (κ,X,Y,obsdim)
102
102
end
103
103
104
- @inline _kernelmatrix (κ:: Kernel ,X,Y,obsdim) = map (x-> kappa (κ,x),pairwise (metric (κ),X,Y,dims= obsdim))
104
+ @inline _kernelmatrix (κ:: SimpleKernel , X, Y, obsdim) =
105
+ map (x -> kappa (κ, x), pairwise (metric (κ), X, Y, dims = obsdim))
105
106
106
107
kernelmatrix (κ:: TransformedKernel , X:: AbstractMatrix , Y:: AbstractMatrix ; obsdim:: Int = defaultobs) =
107
108
kernelmatrix (kernel (κ), apply (κ. transform, X, obsdim = obsdim), apply (κ. transform, Y, obsdim = obsdim), obsdim = obsdim)
@@ -120,9 +121,9 @@ function kerneldiagmatrix(
120
121
)
121
122
@assert obsdim ∈ [1 ,2 ] " obsdim should be 1 or 2 (see docs of kernelmatrix))"
122
123
if obsdim == 1
123
- @compat eachrow (X) .| > x-> _kernel (κ,x, x) # [@views _kernel(κ,X[i,:],X[i,:]) for i in 1:size(X,obsdim)]
124
+ @compat eachrow (X) .| > x-> κ (x, x) # [@views _kernel(κ,X[i,:],X[i,:]) for i in 1:size(X,obsdim)]
124
125
elseif obsdim == 2
125
- @compat eachcol (X) .| > x-> _kernel (κ,x, x) # [@views _kernel(κ,X[:,i],X[:,i]) for i in 1:size(X,obsdim)]
126
+ @compat eachcol (X) .| > x-> κ (x, x) # [@views _kernel(κ,X[:,i],X[:,i]) for i in 1:size(X,obsdim)]
126
127
end
127
128
end
128
129
@@ -133,7 +134,7 @@ In place version of [`kerneldiagmatrix`](@ref)
133
134
"""
134
135
function kerneldiagmatrix! (
135
136
K:: AbstractVector ,
136
- κ:: Kernel ,
137
+ κ:: SimpleKernel ,
137
138
X:: AbstractMatrix ;
138
139
obsdim:: Int = defaultobs
139
140
)
@@ -143,11 +144,11 @@ function kerneldiagmatrix!(
143
144
end
144
145
if obsdim == 1
145
146
for i in eachindex (K)
146
- @inbounds @views K[i] = _kernel (κ, X[i,:],X[i,:])
147
+ @inbounds @views K[i] = κ ( X[i,:], X[i,:])
147
148
end
148
149
else
149
150
for i in eachindex (K)
150
- @inbounds @views K[i] = _kernel (κ, X[:,i],X[:,i])
151
+ @inbounds @views K[i] = κ ( X[:,i], X[:,i])
151
152
end
152
153
end
153
154
return K
0 commit comments