在Julia中評估具有矢量化函數的組合

Question

在Julia中，具有點.的矢量化函數用於按元素操作。

運行f.(x)意味着f(x[1])，f(x[2])，......依次執行

但是，假設我有一個函數，它接受兩個參數，說g(x,y)

我想g(x[1],y[1])，g(x[2],y[1])，g(x[3],y[1])，... ，g(x[1],y[2]),g(x[2],y[2]),g(x[3],y[2])，...

有沒有什麼辦法來評估x和的所有組合？

Answer 1

馬特的答案是好的，但我想提供使用數組理解的選擇：

julia> x = 1:5 
     y = 10:10:50 
     [i + j for i in x, j in y] 
5×5 Array{Int64,2}: 
11 21 31 41 51 
12 22 32 42 52 
13 23 33 43 53 
14 24 34 44 54 
15 25 35 45 55 

在我看來，陣列的理解往往更容易閱讀，比broadcast和reshape更加靈活。

Answer 2

是的，重塑y，使它正交於x。 .矢量化使用broadcast來完成其工作。我把它想象成在所有其他維度上「擠壓」單體尺寸。

這意味着，對於矢量x和y，你可以簡單地重塑他們的一個評價x和y所有組合的產品：

julia> x = 1:5 
     y = 10:10:50 
     (+).(x, reshape(y, 1, length(y))) 
5×5 Array{Int64,2}: 
11 21 31 41 51 
12 22 32 42 52 
13 23 33 43 53 
14 24 34 44 54 
15 25 35 45 55 

請注意，數組的形狀的方向相匹配爭論; x跨越行，並且y跨越列，因爲它被轉置爲單行矩陣。