@@ -179,11 +179,11 @@ vgkldiv(x, y; dims=:) = vvmapreduce((xᵢ, yᵢ) -> xᵢ * (log(xᵢ) - log(yᵢ
179
179
vtgkldiv (x, y; dims= :) = vtmapreduce ((xᵢ, yᵢ) -> xᵢ * (log (xᵢ) - log (yᵢ)) - xᵢ + yᵢ, + , x, y, dims= dims)
180
180
181
181
# Rénya
182
- _vrenyadivergence (p, q, α:: Real , dims) =
182
+ _vrenyidivergence (p, q, α:: Real , dims) =
183
183
vmapreducethen ((pᵢ, qᵢ) -> pᵢ^ α / qᵢ^ (α- 1 ), + , x -> (1 / (α- 1 )) * log (x), p, q, dims= dims)
184
- _vrenyadivergence (p, q, α:: Rational{T} , dims) where {T} = _vrenyadivergence (p, q, float (α), dims)
184
+ _vrenyidivergence (p, q, α:: Rational{T} , dims) where {T} = _vrenyidivergence (p, q, float (α), dims)
185
185
186
- function vrenyadivergence (p, q, α:: Real ; dims= :)
186
+ function vrenyidivergence (p, q, α:: Real ; dims= :)
187
187
if α ≈ 0
188
188
vmapreducethen ((pᵢ, qᵢ) -> ifelse (pᵢ > zero (pᵢ), qᵢ, zero (qᵢ)), + , x -> - log (x), p, q, dims= dims)
189
189
elseif α ≈ 0.5
@@ -196,15 +196,15 @@ function vrenyadivergence(p, q, α::Real; dims=:)
196
196
elseif isinf (α)
197
197
vmapreducethen (/ , max, log, p, q, dims= dims)
198
198
else
199
- _vrenyadivergence (p, q, α, dims)
199
+ _vrenyidivergence (p, q, α, dims)
200
200
end
201
201
end
202
202
203
- _vtrenyadivergence (p, q, α:: Real , dims) =
203
+ _vtrenyidivergence (p, q, α:: Real , dims) =
204
204
vtmapreducethen ((pᵢ, qᵢ) -> pᵢ^ α / qᵢ^ (α- 1 ), + , x -> (1 / (α- 1 )) * log (x), p, q, dims= dims)
205
- _vtrenyadivergence (p, q, α:: Rational{T} , dims) where {T} = _vtrenyadivergence (p, q, float (α), dims)
205
+ _vtrenyidivergence (p, q, α:: Rational{T} , dims) where {T} = _vtrenyidivergence (p, q, float (α), dims)
206
206
207
- function vtrenyadivergence (p, q, α:: Real ; dims= :)
207
+ function vtrenyidivergence (p, q, α:: Real ; dims= :)
208
208
if α ≈ 0
209
209
vtmapreducethen ((pᵢ, qᵢ) -> ifelse (pᵢ > zero (pᵢ), qᵢ, zero (qᵢ)), + , x -> - log (x), p, q, dims= dims)
210
210
elseif α ≈ 0.5
@@ -217,7 +217,7 @@ function vtrenyadivergence(p, q, α::Real; dims=:)
217
217
elseif isinf (α)
218
218
vtmapreducethen (/ , max, log, p, q, dims= dims)
219
219
else
220
- _vtrenyadivergence (p, q, α, dims)
220
+ _vtrenyidivergence (p, q, α, dims)
221
221
end
222
222
end
223
223
0 commit comments