我想知道是否有更快/更好的方法来编写以下 repa 函数
zipOverColumns :: (Source r1 a, Source r2 b)
=> (a -> b -> c)
-> Array r1 DIM1 a
-> Array r2 DIM2 b
-> Array D DIM2 c
zipOverColumns f x y = traverse y id lookup where
lookup get sh@(Z :. _ :. r) = f (x ! (Z :. r)) $ get sh
这是一些示例输出
>>> toList $ zipOverColumns (*) (fromListUnboxed (Z :. 3) [1,2,3])
(fromListUnboxed (Z :. 2 :. 3) [4,5,6,7,8,9])
[4.0,10.0,18.0,7.0,16.0,27.0]
最佳答案
在 Repa 3 上下文中,您的实现是性能最佳的。您会发现以下版本更具可读性:
zipOverColumns2
:: (Source r1 a, Source r2 b)
=> (a -> b -> c)
-> Array r1 DIM1 a
-> Array r2 DIM2 b
-> Array D DIM2 c
zipOverColumns2 f col arr = zipWith f ccol arr
where
ccol = fromFunction (extent arr) (\(Z :. _ :. r) -> col ! (Z :. r))
zipOverColumns2
也比原来快了 15%,但这只是巧合。使用
yarr
函数看起来像:zipOverColumns
:: (USource r1 l1 Dim1 a,
USource r2 l2 Dim2 b, DefaultIFusion r2 l2 fr fl Dim2,
USource fr fl Dim2 c)
=> (a -> b -> c)
-> UArray r1 l1 Dim1 a
-> UArray r2 l2 Dim2 b
-> UArray fr fl Dim2 c
zipOverColumns f col arr = imapM mapF arr
where
mapF (r, _) b = do
a <- col `index` r
return $ f a b
这个版本的速度提高了 30% 到 85%,具体取决于列大小(我已经测试了 4、8 和 1000)。 Gist 用来玩它。
关于haskell - Repa:ZipWith'ing over Columns,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/15595140/