大文字小文字区別を含む関数の型安定度

Question

重心補間式の重みを計算する関数を書いています。タイプの安定性を無視し、それが十分に簡単です：

function baryweights(x) 
    n = length(x) 
    if n == 1; return [1.0]; end # This is obviously not type stable 

    xmin,xmax = extrema(x) 
    x *= 4/(xmax-xmin) 
    #^Multiply by capacity of interval to avoid overflow 
    return [ 
     1/prod(x[i]-x[j] for j in 1:n if j != i) 
     for i = 1:n 
    ] 
end 

タイプの安定性のための問題は、私はn == 1た場合の正しい型の配列を返すことができるようにn > 1例戻り値の型をうまくすることです。これを達成するための簡単なトリックはありますか？

Answer 1

は単に仮引数に再帰関数を呼び出す：

function baryweights(x) 
    n = length(x) 
    if n == 1 
     T = eltype(baryweights(zeros(eltype(x),2))) 
     return [one(T)] 
    end 

    xmin,xmax = extrema(x) 
    let x = 4/(xmax-xmin) * x 
     #^Multiply by capacity of interval to avoid overflow, 
     # and wrap in let to avoid another source of type instability 
     # (https://github.com/JuliaLang/julia/issues/15276) 
     return [ 
      1/prod(x[i]-x[j] for j in 1:n if j != i) 
      for i = 1:n 
     ] 
    end 
end 

Answer 2

私は私はあなたの計画を理解している場合わかりません。しかし、おそらくこのような何かが助けることができますか？ - >

baryone(t::T) where T<:Real = [1.] 
baryone(t::T) where T<:Complex = [1im] # or whatever you like here 

function baryweights(x::Array{T,1}) where T<:Number 
    n = length(x) 
    n == 1 && return baryone(x[1]) 
    xmin,xmax = extrema(x) # don't forget fix extrema for complex! :) 
    x *= 4/(xmax-xmin) 
    #^Multiply by capacity of interval to avoid overflow 
    return [ 
     1/prod(x[i]-x[j] for j in 1:n if j != i) 
     for i = 1:n 
    ] 
end 

警告：私はまだ初心者です！私が試してみたら@code_warntype baryweights([1])私はちょうどたくさんの警告を見る。 （もし私がbaryoneに電話するのを避けても）。たとえば、nはAnyです。

function baryweights(x::Array{T,1}) where T<:Number 
    n = length(x) 
    n == 1 && return baryone(x[1]) 
    xmin,xmax = extrema(x) # don't forget fix extrema for complex! :) 
    let y = x * 4/(xmax-xmin) 
     #^Multiply by capacity of interval to avoid overflow 
     return [ 
      1/prod(y[i]-y[j] for j in 1:n if j != i) 
      for i = 1:n 
     ] 
    end 
end 

EDIT2：

編集：私たちは別の変数（y）を使用している場合、今 I asked on discourseとは、その@code_warnリターンより良い結果を参照してくださいCore.Box EDになるようにyを避けるために、私はletを追加しまし