为许多相似功能实现多重分派的有效方法
我正在编写一些软件,其中涉及一个数量的各种功能形式的库。我想利用 Julia 的多重分派,但想知道是否有更有效的方法来实现此过程。
例如,考虑一个包含以下两个函数的库
function firstfunction(x::Float64)
return 2*x
end
function secondfunction(x::Float64)
return x^2
end
我还想实现多个调度方法,这些方法可以将这些函数形式应用于值向量或向量数组(矩阵)。我可以这样做
function firstfunction(x::Float64)
return 2*x
end
function firstfunction(xs::Vector{Float64})
f = similar(xs)
for i = 1:size(xs, 1)
f[i] = 2*xs[i]
end
return f
end
function firstfunction(xss::Matrix{Float64})
f = similar(xss)
for i = 1:size(xss, 1)
for j = 1:size(xss, 2)
f[i, j] = 2*xss[i, j]
end
return f
end
function secondfunction(x::Float64)
return x^2
end
function secondfunction(xs::Vector{Float64})
f = similar(xs)
for i = 1:size(xs, 1)
f[i] = xs[i]^2
end
return f
end
function secondfunction(xss::Matrix{Float64})
f = similar(xss)
for i = 1:size(xss, 1)
for j = 1:size(xss, 2)
f[i, j] = xss[i, j]^2
end
return f
end
但是由于该函数的所有三个版本都使用相同的内核,并且各种分派的操作在所有函数形式中都是相同的,我想知道是否有更有效的方法来编写它,以便为库(例如thirdfunction)只涉及显式编写内核函数,而不必为库中的函数形式键入2*n本质上相同的函数n。
回答
做就是了:
function thirdfunction(x::Union{Number, Array{<:Number}})
return x.^0.5
end
这就是 Julia 中多次分派的美妙之处:
julia> thirdfunction(4)
2.0
julia> thirdfunction([4,9])
2-element Array{Float64,1}:
2.0
3.0
julia> thirdfunction([4 9; 16 25])
2×2 Array{Float64,2}:
2.0 3.0
4.0 5.0
请注意,但是在您的情况下,只有一个函数的单一表示并让用户决定使用点运算符 ( .)对其进行矢量化可能是有意义的。
function fourthfunction(x::Real)
min(x, 5)
end
现在用户只需要在需要时添加一个点:
julia> fourthfunction(4)
4
julia> fourthfunction.([4,9])
2-element Array{Int64,1}:
4
5
julia> fourthfunction.([4 9; 16 25])
2×2 Array{Int64,2}:
4 5
5 5
由于 Julia 中的矢量化非常简单,因此您应该尽可能考虑这种设计,