Data.Array.Accelerate.Utility.Arrange

Documentation

mapWithIndex :: (Shape sh, Elt a, Elt b) => (Exp sh -> Exp a -> Exp b) -> Acc (Array sh a) -> Acc (Array sh b)

gather :: (Shape ix, Shape ix', Elt ix', Elt a) => Acc (Array ix ix') -> Acc (Array ix' a) -> Acc (Array ix a)

scatter :: (Shape ix, Shape ix', Elt ix', Elt a) => (Exp a -> Exp a -> Exp a) -> Acc (Array ix ix') -> Acc (Array ix' a) -> Acc (Array ix a) -> Acc (Array ix' a)