대소 문자 구별이 포함 된 함수의 유형 안정성

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 에드로 y을 피하기 위해 내가 let 추가를