julia - 为许多类似功能实现多次调度的有效方法

Question

我正在编写一些软件，其中涉及一个数量的各种函数形式的库。我想利用 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 的功能基本相同库中的功能形式。

Answer 1

做就是了:

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 中的矢量化非常简单，因此您应该尽可能考虑这种设计，