| Safe Haskell | None |
|---|
Feldspar.Repa
- data Z = Z
- data tail :. head = tail :. head
- type DIM0 = Z
- type DIM1 = DIM0 :. Data Length
- type DIM2 = DIM1 :. Data Length
- type DIM3 = DIM2 :. Data Length
- class Shape sh where
- data All = All
- data Any sh = Any
- type family FullShape ss
- type family SliceShape ss
- class Slice ss where
- sliceOfFull :: ss -> FullShape ss -> SliceShape ss
- fullOfSlice :: ss -> SliceShape ss -> FullShape ss
- data Vector sh a = Vector sh (sh -> a)
- type DVector sh a = Vector sh (Data a)
- fromVector :: (Shape sh, Type a) => DVector sh a -> Data [a]
- toVector :: (Shape sh, Type a) => sh -> Data [a] -> DVector sh a
- freezeVector :: (Shape sh, Type a) => DVector sh a -> (Data [Length], Data [a])
- fromList :: Type a => [Data a] -> Data [a]
- thawVector :: (Shape sh, Type a) => (Data [Length], Data [a]) -> DVector sh a
- memorize :: (Shape sh, Type a) => DVector sh a -> DVector sh a
- extent :: Vector sh a -> sh
- newExtent :: sh -> Vector sh a -> Vector sh a
- traverse :: (Shape sh, Shape sh') => Vector sh a -> (sh -> sh') -> ((sh -> a) -> sh' -> a') -> Vector sh' a'
- replicate :: (Slice sl, Shape (FullShape sl), Shape (SliceShape sl)) => sl -> Vector (SliceShape sl) a -> Vector (FullShape sl) a
- slice :: (Slice sl, Shape (FullShape sl), Shape (SliceShape sl)) => Vector (FullShape sl) a -> sl -> Vector (SliceShape sl) a
- reshape :: (Shape sh, Shape sh') => sh -> Vector sh' a -> Vector sh a
- unit :: a -> Vector Z a
- (!:) :: Shape sh => Vector sh a -> sh -> a
- diagonal :: Vector DIM2 a -> Vector DIM1 a
- backpermute :: (Shape sh, Shape sh') => sh' -> (sh' -> sh) -> Vector sh a -> Vector sh' a
- map :: (a -> b) -> Vector sh a -> Vector sh b
- zip :: Shape sh => Vector sh a -> Vector sh b -> Vector sh (a, b)
- zipWith :: Shape sh => (a -> b -> c) -> Vector sh a -> Vector sh b -> Vector sh c
- fold :: (Shape sh, Syntax a) => (a -> a -> a) -> a -> Vector (sh :. Data Length) a -> Vector sh a
- fold' :: (Shape sh, Syntax a) => (a -> a -> a) -> Vector sh a -> Vector (sh :. Data Length) a -> Vector sh a
- sum :: (Shape sh, Type a, Numeric a) => DVector (sh :. Data Length) a -> DVector sh a
- (...) :: Data Index -> Data Index -> DVector DIM1 Index
- stencil :: DVector DIM2 Float -> DVector DIM2 Float
- laplace :: Data Length -> DVector DIM2 Float -> DVector DIM2 Float
- transpose2D :: Vector DIM2 e -> Vector DIM2 e
- mmMult :: (Type e, Numeric e) => DVector DIM2 e -> DVector DIM2 e -> DVector DIM2 e
- mapDIM1 :: (Data Index -> Data Index) -> DIM1 -> DIM1
- indexed :: Data Length -> (Data Index -> a) -> Vector DIM1 a
- length :: Vector DIM1 a -> Data Length
- newLen :: Syntax a => Data Length -> Vector DIM1 a -> Vector DIM1 a
- (++) :: Syntax a => Vector DIM1 a -> Vector DIM1 a -> Vector DIM1 a
- take :: Data Length -> Vector DIM1 a -> Vector DIM1 a
- drop :: Data Length -> Vector DIM1 a -> Vector DIM1 a
- splitAt :: Data Index -> Vector DIM1 a -> (Vector DIM1 a, Vector DIM1 a)
- head :: Syntax a => Vector DIM1 a -> a
- last :: Syntax a => Vector DIM1 a -> a
- tail :: Vector DIM1 a -> Vector DIM1 a
- init :: Vector DIM1 a -> Vector DIM1 a
- tails :: Vector DIM1 a -> Vector DIM1 (Vector DIM1 a)
- inits :: Vector DIM1 a -> Vector DIM1 (Vector DIM1 a)
- inits1 :: Vector DIM1 a -> Vector DIM1 (Vector DIM1 a)
- permute :: (Data Length -> Data Index -> Data Index) -> Vector DIM1 a -> Vector DIM1 a
- reverse :: Syntax a => Vector DIM1 a -> Vector DIM1 a
- rotateVecL :: Syntax a => Data Index -> Vector DIM1 a -> Vector DIM1 a
- rotateVecR :: Syntax a => Data Index -> Vector DIM1 a -> Vector DIM1 a
- replicate1 :: Data Length -> a -> Vector DIM1 a
- enumFromTo :: Data Index -> Data Index -> Vector DIM1 (Data Index)
- enumFrom :: Data Index -> Vector DIM1 (Data Index)
- unzip :: Vector DIM1 (a, b) -> (Vector DIM1 a, Vector DIM1 b)
- foldl :: Syntax a => (a -> b -> a) -> a -> Vector DIM1 b -> a
- fold1 :: Syntax a => (a -> a -> a) -> Vector DIM1 a -> a
- sum1 :: (Syntax a, Num a) => Vector DIM1 a -> a
- maximum :: Ord a => Vector DIM1 (Data a) -> Data a
- minimum :: Ord a => Vector DIM1 (Data a) -> Data a
- scalarProd :: (Syntax a, Num a) => Vector DIM1 a -> Vector DIM1 a -> a
- tVec :: Patch a a -> Patch (Vector DIM1 a) (Vector DIM1 a)
- tVec1 :: Patch a a -> Patch (Vector DIM1 (Data a)) (Vector DIM1 (Data a))
- tVec2 :: Patch a a -> Patch (Vector DIM2 (Data a)) (Vector DIM2 (Data a))
Documentation
Constructors
| tail :. head |
Methods
Get the number of dimensions in a shape
The shape of an array of size zero, with a particular dimension
The shape of an array with size one, with a particular dimension
size :: sh -> Data LengthSource
Get the total number of elements in an array with this shape.
toIndex :: sh -> sh -> Data IndexSource
Index into flat, linear, row-major representation
fromIndex :: sh -> Data Index -> shSource
Inverse of toIndex.
intersectDim :: sh -> sh -> shSource
The intersection of two dimensions.
inRange :: sh -> sh -> sh -> Data BoolSource
Check whether an index is within a given shape.
inRange l u i checks that i fits between l and u.
toList :: sh -> [Data Length]Source
Turn a shape into a list. Used in the Syntactic instance.
toShape :: Int -> Data [Length] -> shSource
Reconstruct a shape. Used in the Syntactic instance.
type family SliceShape ss Source
Methods
sliceOfFull :: ss -> FullShape ss -> SliceShape ssSource
fullOfSlice :: ss -> SliceShape ss -> FullShape ssSource
- Vectors
Constructors
| Vector sh (sh -> a) |
fromVector :: (Shape sh, Type a) => DVector sh a -> Data [a]Source
- Fuctions
Store a vector in an array.
toVector :: (Shape sh, Type a) => sh -> Data [a] -> DVector sh aSource
Restore a vector from an array
memorize :: (Shape sh, Type a) => DVector sh a -> DVector sh aSource
Store a vector in memory. Use this function instead of force if
possible as it is both much more safe and faster.
newExtent :: sh -> Vector sh a -> Vector sh aSource
Change the extent of the vector to the supplied value. If the supplied extent will contain more elements than the old extent, the new elements will have undefined value.
traverse :: (Shape sh, Shape sh') => Vector sh a -> (sh -> sh') -> ((sh -> a) -> sh' -> a') -> Vector sh' a'Source
Change shape and transform elements of a vector. This function is the most general way of manipulating a vector.
replicate :: (Slice sl, Shape (FullShape sl), Shape (SliceShape sl)) => sl -> Vector (SliceShape sl) a -> Vector (FullShape sl) aSource
Duplicates part of a vector along a new dimension.
slice :: (Slice sl, Shape (FullShape sl), Shape (SliceShape sl)) => Vector (FullShape sl) a -> sl -> Vector (SliceShape sl) aSource
Extracts a slice from a vector.
reshape :: (Shape sh, Shape sh') => sh -> Vector sh' a -> Vector sh aSource
Change the shape of a vector. This function is potentially unsafe, the new shape need to have fewer or equal number of elements compared to the old shape.
backpermute :: (Shape sh, Shape sh') => sh' -> (sh' -> sh) -> Vector sh a -> Vector sh' aSource
Change the shape of a vector.
zip :: Shape sh => Vector sh a -> Vector sh b -> Vector sh (a, b)Source
Combines the elements of two vectors. The size of the resulting vector will be the intersection of the two argument vectors.
zipWith :: Shape sh => (a -> b -> c) -> Vector sh a -> Vector sh b -> Vector sh cSource
Combines the elements of two vectors pointwise using a function. The size of the resulting vector will be the intersection of the two argument vectors.
fold :: (Shape sh, Syntax a) => (a -> a -> a) -> a -> Vector (sh :. Data Length) a -> Vector sh aSource
Reduce a vector along its last dimension
fold' :: (Shape sh, Syntax a) => (a -> a -> a) -> Vector sh a -> Vector (sh :. Data Length) a -> Vector sh aSource
A generalization of fold which allows for different initial
values when starting to fold.
sum :: (Shape sh, Type a, Numeric a) => DVector (sh :. Data Length) a -> DVector sh aSource
Summing a vector along its last dimension
mmMult :: (Type e, Numeric e) => DVector DIM2 e -> DVector DIM2 e -> DVector DIM2 eSource
Matrix multiplication
Operations on one dimensional vectors
newLen :: Syntax a => Data Length -> Vector DIM1 a -> Vector DIM1 aSource
Change the length of the vector to the supplied value. If the supplied length is greater than the old length, the new elements will have undefined value. The resulting vector has only one segment.
permute :: (Data Length -> Data Index -> Data Index) -> Vector DIM1 a -> Vector DIM1 aSource
Permute a vector
enumFromTo :: Data Index -> Data Index -> Vector DIM1 (Data Index)Source
enumFromTo m n: Enumerate the indexes from m to n
In order to enumerate a different type, use i2n, e.g:
map i2n (10...20) :: Vector1 Word8
enumFrom :: Data Index -> Vector DIM1 (Data Index)Source
enumFrom m: Enumerate the indexes from m to maxBound
foldl :: Syntax a => (a -> b -> a) -> a -> Vector DIM1 b -> aSource
Corresponds to the standard foldl.