Safe Haskell | None |
---|
Documentation
data Array sh a
mapAccumLSimple :: (C sh, MV sh, C n, MV n, C acc, C a, C a, C b, C b) => (Exp acc -> Exp a -> Exp (acc, b)) -> Array p sh acc -> Array p (sh, n) a -> IO (p -> IO (Array (sh, n) b))Source
foldOuterL :: (C sh, MV sh, C n, MV n, C a, C a) => (Exp a -> Exp b -> Exp a) -> Array p sh a -> Array p (n, sh) b -> IO (p -> IO (Array sh a))Source
scatter :: (C sh0, Index sh0 ~ ix0, C sh1, Index sh1 ~ ix1, MV sh1, C a, C a) => (Exp a -> Exp a -> Exp a) -> Array p sh1 a -> Array p sh0 (ix1, a) -> IO (p -> IO (Array sh1 a))Source