arrays - 将包含输入参数的数组传递给 Julia 中的函数

Question

如何将包含所需维度的数组传递给 Julia 中的 ones 或 rand 等函数？

例如，我想执行

dims = [3 4]
rand(dims)

然后收到类似的东西

3×4 Array{Float64,2}:
 0.300811  0.140124  0.756915   0.268328 
 0.732461  0.900773  0.251334   0.0338452
 0.54227   0.439385  0.0812104  0.612996 

作为输出。

执行第一个代码块会返回从数组 dimen = [3 4] 中随机选择的单个数字，但是 rand(dims = dimen) 会返回一个错误:函数 rand 不接受关键字参数错误。

有没有办法做我希望做的事？

Answer 1

您正在寻找的是 "splat" operator ... , 作为 described in the documentation :

On the flip side, it is often handy to "splat" the values contained in an iterable collection into a function call as individual arguments. To do this, one also uses ... but in the function call instead

在你的情况下:

julia> dims = [3 4];
julia> rand(dims...)

3×4 Array{Float64,2}:
 0.664496  0.190208  0.167208  0.172296
 0.632465  0.374373  0.417636  0.354948
 0.743741  0.740435  0.602339  0.401814