ベクタライズ特定の引数

Question

上の機能は、私は関数に

myfunc(a, x::Int64) = a * x 

があると私は

myfunc{N}(a, x::Array{Int64, N}) = map(x -> myfunc(a, x), x) 

のようなものを持っているように、私はマクロがある知っている、唯一の第二引数をベクトル化したいです@ vectorize_1argまたは@ vectorize_2arg。 しかし、それらのマクロはすべての引数をベクトル化します。

質問：特定の引数で関数を便利にベクトル化するにはどうすればよいですか？上記の例のようにハードコードする必要がありますか？

Answer 1

あなたが唯一の第二引数のベクトル化をしたい機能を拡張するために探しているなら、これはそれを行う必要があります。

このように使用

macro vectorize_on_2nd(S, f) 
    S = esc(S); f = esc(f); N = esc(:N) 
    quote 
     ($f){$N}(a, x::AbstractArray{$S, $N}) = 
      reshape([($f)(a, x[i]) for i in eachindex(x)], size(x)) 
    end 
end 

：

あなた myfunc{N}(::Any, ::AbstractArray{Int64,N})を与える必要があります

@vectorize_on_2nd Int64 myfunc 

方法。