为任何整数数组定义函数

Question

我想定义一个函数，该函数将任何以整数（且只有整数）为元素的2维数组作为输入。尽管我知道我不必在Julia中指定函数的参数类型，但我想这样做是为了加快速度。

使用类型层次结构，我可以使用以下代码对将整数作为输入的函数执行此操作：

julia> function sum_two(x::Integer)
           return x+2
       end
sum_two (generic function with 1 method)

julia> sum_two(Int8(4))
6

julia> sum_two(Int16(4))

但是，当我尝试使用该类型时Array{Integer,2}，出现以下错误：

julia> function sum_array(x::Array{Integer,2})
           return sum(x)
       end
sum_array (generic function with 1 method)

julia> sum_array(ones(Int8,10,10))
ERROR: MethodError: no method matching sum_array(::Array{Int8,2})
Closest candidates are:
  sum_array(::Array{Integer,2}) at REPL[4]:2
Stacktrace:
 [1] top-level scope at none:0

我没有办法解决这个问题。一种选择是通过以下方式为Integer的每个最低级别的子类型定义方法：

function sum_array(x::Array{Int8,2})
           return sum(x)
       end

function sum_array(x::Array{UInt8,2})
           return sum(x)
       end
.
.
.

但这看起来并不实用。

Answer 1

首先：指定函数输入参数的类型不会加快代码的速度。这是一个误会。定义结构时，应指定具体的字段类型，但是对于函数签名，它对性能没有任何影响。您可以使用它来控制调度。

现在，你的问题：Julia的类型参数是不变的，也就是说即使S<:T是真实的，A{S}<:A{T}是不正确的。您可以在这里阅读有关此内容的更多信息：https : //docs.julialang.org/en/v1/manual/types/index.html#Parametric-Composite-Types-1

因此，ones(Int8,10,10)，这是一个Matrix{Int8}被不的子类型Matrix{Integer}。

要使代码正常工作，您可以执行以下操作：

function sum_array(x::Array{T, 2}) where {T<:Integer}
    return sum(x)
end

或使用这个不错的捷径

function sum_array(x::Array{<:Integer, 2})
    return sum(x)
end