我想定义一个函数,它将任何以整数(并且只有整数)作为其元素的 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
.
.
.
但是看起来不太实用。
最佳答案
首先:为函数指定输入参数的类型不会加速代码。这是一种误解。您应该在定义结构体时指定具体的字段类型,但对于函数签名,它对性能没有任何影响。你用它来控制调度。
现在,对于您的问题: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
关于arrays - 为任何整数数组定义函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57279788/