haskell - Repa 中的基本顺序数组转换

Question

我想知道repa中是否有(//)的类似物？

无法并行化的数组转换需要它。例如，如果函数需要整个数组来更改数组的单个条目，然后将其应用于新数组等等(并且它应该按顺序运行)。

Answer 1

(//)可以用Data.Array.Repa.fromFunction来实现:

import Data.Array.Repa

(//) :: Shape sh => Array sh a -> [(sh,a)] -> Array sh a
(//) arr us = fromFunction (extent arr) (\sh -> case lookup sh us of
                                                 Just a  -> a
                                                 Nothing -> index arr sh)

fromFunction 可以传递一个 Shape sh => s -> a 类型的函数，它本身可以利用整个数组。

上述实现一次执行所有更新。