矢量化计算
在对数组进行操作时,经常遇到要对其内部元素逐一实施同一种计算的情况。例如,对数组中的每一个元素求绝对值、平方;或以元素为参数调用某个函数,并将计算结果生成新的数组,等等。常规的做法是遍历数组,对每个元素单独实现计算过程。但在密集数值计算的应用中,这种逐元计算的需要往往会频繁出现,为此,Julia 提供了矢量化计算机制,能够简洁地实现逐元计算,而且无须开发者实现循环遍历操作。
map 函数
在 Julia 中,常规矢量化的方法是使用 map 函数。
map(f::Function, c...) -> collection
其中,参数 f 是一个函数对象, c 是一个可变参数的迭代数集组。调用时, map 内部会自动对 c 中的每个元素执行 f 函数,并将计算结果组建为新的数集。
可见, map 函数也适用于元组、Range 等类型。甚至还可以更为灵活地使用生成器。
需要注意的是,生成器需要使用小括号来界定,否则会报错。
一般情况下, map 函数生成的结果数集结构总是与输入的数集一致,即:输入数集是元组,则结果类型也是元组;如果输入数集是数组,结果数集也是数组。
函数对象参数除了是已有的函数外,也可以是匿名函数。
像上面这种匿名函数略显复杂,我们可以使用 do 代码块实现。
另外作为函数对象的一种,运算符也可以参与 map 函数计算。
可见, map 函数支持可变参数,只需按照函数对象参数的需求提供对应的参数即可。