Data.Array.Repa.Operators.Mapping

Generic maps

map :: (Shape sh, Source r a) => (a -> b) -> Array r sh a -> Array D sh b

Apply a worker function to each element of an array, yielding a new array with the same extent.

zipWith :: (Shape sh, Source r1 a, Source r2 b) => (a -> b -> c) -> Array r1 sh a -> Array r2 sh b -> Array D sh c

Combine two arrays, element-wise, with a binary operator. If the extent of the two array arguments differ, then the resulting array's extent is their intersection.

(+^) :: (Shape sh, Source r1 c, Source r2 c, Num c) => Array r1 sh c -> Array r2 sh c -> Array D sh c infixl 6

(-^) :: (Shape sh, Source r1 c, Source r2 c, Num c) => Array r1 sh c -> Array r2 sh c -> Array D sh c infixl 6

(*^) :: (Shape sh, Source r1 c, Source r2 c, Num c) => Array r1 sh c -> Array r2 sh c -> Array D sh c infixl 7

(/^) :: (Shape sh, Source r1 c, Source r2 c, Fractional c) => Array r1 sh c -> Array r2 sh c -> Array D sh c infixl 7

Structured maps

class Structured r1 a b where

Structured versions of map and zipWith that preserve the representation of cursored and partitioned arrays.

For cursored (C) arrays, the cursoring of the source array is preserved.

For partitioned (P) arrays, the worker function is fused with each array partition separately, instead of treating the whole array as a single bulk object.

Preserving the cursored and/or paritioned representation of an array is will make follow-on computation more efficient than if the array was converted to a vanilla Delayed (D) array as with plain map and zipWith.

If the source array is not cursored or partitioned then smap and szipWith are identical to the plain functions.

Associated Types

type TR r1

The target result representation.

Methods

smap :: Shape sh => (a -> b) -> Array r1 sh a -> Array (TR r1) sh b

Structured map.

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array r1 sh a -> Array (TR r1) sh b

Structured zipWith. If you have a cursored or partitioned source array then use that as the third argument (corresponding to r1 here)

Instances

Structured D a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR D

Methods

smap :: Shape sh => (a -> b) -> Array D sh a -> Array (TR D) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array D sh a -> Array (TR D) sh b

Storable a => Structured F a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR F

Methods

smap :: Shape sh => (a -> b) -> Array F sh a -> Array (TR F) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array F sh a -> Array (TR F) sh b

Structured B Word8 b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR B

Methods

smap :: Shape sh => (Word8 -> b) -> Array B sh Word8 -> Array (TR B) sh b

szipWith :: (Shape sh, Source r c) => (c -> Word8 -> b) -> Array r sh c -> Array B sh Word8 -> Array (TR B) sh b

Structured X a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR X

Methods

smap :: Shape sh => (a -> b) -> Array X sh a -> Array (TR X) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array X sh a -> Array (TR X) sh b

Structured C a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR C

Methods

smap :: Shape sh => (a -> b) -> Array C sh a -> Array (TR C) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array C sh a -> Array (TR C) sh b

Unbox a => Structured U a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR U

Methods

smap :: Shape sh => (a -> b) -> Array U sh a -> Array (TR U) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array U sh a -> Array (TR U) sh b

Structured r1 a b => Structured (S r1) a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR (S r1)

Methods

smap :: Shape sh => (a -> b) -> Array (S r1) sh a -> Array (TR (S r1)) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array (S r1) sh a -> Array (TR (S r1)) sh b

Structured r1 a b => Structured (I r1) a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR (I r1)

Methods

smap :: Shape sh => (a -> b) -> Array (I r1) sh a -> Array (TR (I r1)) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array (I r1) sh a -> Array (TR (I r1)) sh b

(Structured r1 a b, Structured r2 a b) => Structured (P r1 r2) a b

Defined in Data.Array.Repa.Operators.Mapping

Associated Types

type TR (P r1 r2)

Methods

smap :: Shape sh => (a -> b) -> Array (P r1 r2) sh a -> Array (TR (P r1 r2)) sh b

szipWith :: (Shape sh, Source r c) => (c -> a -> b) -> Array r sh c -> Array (P r1 r2) sh a -> Array (TR (P r1 r2)) sh b