Numeric.LinearAlgebra.Devel

Contents

• FFI tools
• ST
  • Mutable Vectors
  • Mutable Matrices
  • Unsafe functions
• Special maps and zips
• Sparse representation
• Misc

Description

Synopsis

• createVector :: Storable a => Int -> IO (Vector a)
• createMatrix :: Storable a => MatrixOrder -> Int -> Int -> IO (Matrix a)
• class TransArray c where
  • type Trans c b
  • type TransRaw c b
• data MatrixOrder
  • = RowMajor
  • | ColumnMajor
• orderOf :: Matrix t -> MatrixOrder
• cmat :: Element t => Matrix t -> Matrix t
• fmat :: Element t => Matrix t -> Matrix t
• matrixFromVector :: Storable t => MatrixOrder -> Int -> Int -> Vector t -> Matrix t
• unsafeFromForeignPtr :: Storable a => ForeignPtr a -> Int -> Int -> Vector a
• unsafeToForeignPtr :: Storable a => Vector a -> (ForeignPtr a, Int, Int)
• check :: String -> IO CInt -> IO ()
• (//) :: x -> (x -> y) -> y
• (#|) :: IO CInt -> String -> IO ()
• at' :: Storable a => Vector a -> Int -> a
• atM' :: Storable t => Matrix t -> Int -> Int -> t
• fi :: Int -> CInt
• ti :: CInt -> Int
• data STVector s t
• newVector :: Storable t => t -> Int -> ST s (STVector s t)
• thawVector :: Storable t => Vector t -> ST s (STVector s t)
• freezeVector :: Storable t => STVector s t -> ST s (Vector t)
• runSTVector :: Storable t => (forall s. ST s (STVector s t)) -> Vector t
• readVector :: Storable t => STVector s t -> Int -> ST s t
• writeVector :: Storable t => STVector s t -> Int -> t -> ST s ()
• modifyVector :: Storable t => STVector s t -> Int -> (t -> t) -> ST s ()
• liftSTVector :: Storable t => (Vector t -> a) -> STVector s t -> ST s a
• data STMatrix s t
• newMatrix :: Storable t => t -> Int -> Int -> ST s (STMatrix s t)
• thawMatrix :: Element t => Matrix t -> ST s (STMatrix s t)
• freezeMatrix :: Element t => STMatrix s t -> ST s (Matrix t)
• runSTMatrix :: Storable t => (forall s. ST s (STMatrix s t)) -> Matrix t
• readMatrix :: Storable t => STMatrix s t -> Int -> Int -> ST s t
• writeMatrix :: Storable t => STMatrix s t -> Int -> Int -> t -> ST s ()
• modifyMatrix :: Storable t => STMatrix s t -> Int -> Int -> (t -> t) -> ST s ()
• liftSTMatrix :: Element t => (Matrix t -> a) -> STMatrix s t -> ST s a
• mutable :: Element t => (forall s. (Int, Int) -> STMatrix s t -> ST s u) -> Matrix t -> (Matrix t, u)
• extractMatrix :: Element a => STMatrix s1 a -> RowRange -> ColRange -> ST s2 (Matrix a)
• setMatrix :: Element t => STMatrix s t -> Int -> Int -> Matrix t -> ST s ()
• rowOper :: (Num t, Element t) => RowOper t -> STMatrix s t -> ST s ()
• data RowOper t
  • = AXPY t Int Int ColRange
  • | SCAL t RowRange ColRange
  • | SWAP Int Int ColRange
• data RowRange
  • = AllRows
  • | RowRange Int Int
  • | Row Int
  • | FromRow Int
• data ColRange
  • = AllCols
  • | ColRange Int Int
  • | Col Int
  • | FromCol Int
• gemmm :: Element t => t -> Slice s t -> t -> Slice s t -> Slice s t -> ST s ()
• data Slice s t = Slice (STMatrix s t) Int Int Int Int
• newUndefinedVector :: Storable t => Int -> ST s (STVector s t)
• unsafeReadVector :: Storable t => STVector s t -> Int -> ST s t
• unsafeWriteVector :: Storable t => STVector s t -> Int -> t -> ST s ()
• unsafeThawVector :: Storable t => Vector t -> ST s (STVector s t)
• unsafeFreezeVector :: Storable t => STVector s t -> ST s (Vector t)
• newUndefinedMatrix :: Storable t => MatrixOrder -> Int -> Int -> ST s (STMatrix s t)
• unsafeReadMatrix :: Storable t => STMatrix s t -> Int -> Int -> ST s t
• unsafeWriteMatrix :: Storable t => STMatrix s t -> Int -> Int -> t -> ST s ()
• unsafeThawMatrix :: Storable t => Matrix t -> ST s (STMatrix s t)
• unsafeFreezeMatrix :: Storable t => STMatrix s t -> ST s (Matrix t)
• mapVectorWithIndex :: (Storable a, Storable b) => (Int -> a -> b) -> Vector a -> Vector b
• zipVector :: (Storable a, Storable b, Storable (a, b)) => Vector a -> Vector b -> Vector (a, b)
• zipVectorWith :: (Storable a, Storable b, Storable c) => (a -> b -> c) -> Vector a -> Vector b -> Vector c
• unzipVector :: (Storable a, Storable b, Storable (a, b)) => Vector (a, b) -> (Vector a, Vector b)
• unzipVectorWith :: (Storable (a, b), Storable c, Storable d) => ((a, b) -> (c, d)) -> Vector (a, b) -> (Vector c, Vector d)
• mapVectorM :: (Storable a, Storable b, Monad m) => (a -> m b) -> Vector a -> m (Vector b)
• mapVectorM_ :: (Storable a, Monad m) => (a -> m ()) -> Vector a -> m ()
• mapVectorWithIndexM :: (Storable a, Storable b, Monad m) => (Int -> a -> m b) -> Vector a -> m (Vector b)
• mapVectorWithIndexM_ :: (Storable a, Monad m) => (Int -> a -> m ()) -> Vector a -> m ()
• foldLoop :: (Int -> t -> t) -> t -> Int -> t
• foldVector :: Storable a => (a -> b -> b) -> b -> Vector a -> b
• foldVectorG :: Storable t1 => (Int -> (Int -> t1) -> t2 -> t2) -> t2 -> Vector t1 -> t2
• foldVectorWithIndex :: Storable a => (Int -> a -> b -> b) -> b -> Vector a -> b
• mapMatrixWithIndex :: (Element a, Storable b) => ((Int, Int) -> a -> b) -> Matrix a -> Matrix b
• mapMatrixWithIndexM :: (Element a, Storable b, Monad m) => ((Int, Int) -> a -> m b) -> Matrix a -> m (Matrix b)
• mapMatrixWithIndexM_ :: (Element a, Num a, Monad m) => ((Int, Int) -> a -> m ()) -> Matrix a -> m ()
• liftMatrix :: (Element a, Element b) => (Vector a -> Vector b) -> Matrix a -> Matrix b
• liftMatrix2 :: (Element t, Element a, Element b) => (Vector a -> Vector b -> Vector t) -> Matrix a -> Matrix b -> Matrix t
• liftMatrix2Auto :: (Element t, Element a, Element b) => (Vector a -> Vector b -> Vector t) -> Matrix a -> Matrix b -> Matrix t
• data CSR = CSR {
  • csrVals :: Vector Double
  • csrCols :: Vector CInt
  • csrRows :: Vector CInt
  • csrNRows :: Int
  • csrNCols :: Int
  }
• fromCSR :: CSR -> GMatrix
• mkCSR :: AssocMatrix -> CSR
• data GMatrix
  • = SparseR {
    • gmCSR :: CSR
    • nRows :: Int
    • nCols :: Int
    }
  • | SparseC {
    • gmCSC :: CSC
    • nRows :: Int
    • nCols :: Int
    }
  • | Diag {
    • diagVals :: Vector Double
    • nRows :: Int
    • nCols :: Int
    }
  • | Dense {
    • gmDense :: Matrix Double
    • nRows :: Int
    • nCols :: Int
    }
• toByteString :: Storable t => Vector t -> ByteString
• fromByteString :: Storable t => ByteString -> Vector t
• showInternal :: Storable t => Matrix t -> IO ()
• reorderVector :: Element a => Vector CInt -> Vector CInt -> Vector a -> Vector a

FFI tools

ST

Mutable Vectors

Mutable Matrices

Unsafe functions

Special maps and zips

>>> mapMatrixWithIndex (\(i,j) v -> 100*v + 10*fromIntegral i + fromIntegral j) (ident 3:: Matrix Double)
(3><3)
 [ 100.0,   1.0,   2.0
 ,  10.0, 111.0,  12.0
 ,  20.0,  21.0, 122.0 ]

>>> mapMatrixWithIndexM (\(i,j) v -> Just $ 100*v + 10*fromIntegral i + fromIntegral j) (ident 3:: Matrix Double)
Just (3><3)
 [ 100.0,   1.0,   2.0
 ,  10.0, 111.0,  12.0
 ,  20.0,  21.0, 122.0 ]

>>> mapMatrixWithIndexM_ (\(i,j) v -> printf "m[%d,%d] = %.f\n" i j v :: IO()) ((2><3)[1 :: Double ..])
m[0,0] = 1
m[0,1] = 2
m[0,2] = 3
m[1,0] = 4
m[1,1] = 5
m[1,2] = 6

Sparse representation

>>> let m = mkSparse [((0,999),1.0),((1,1999),2.0)]
>>> m
SparseR {gmCSR = CSR {csrVals = fromList [1.0,2.0],
                      csrCols = fromList [1000,2000],
                      csrRows = fromList [1,2,3],
                      csrNRows = 2,
                      csrNCols = 2000},
                      nRows = 2,
                      nCols = 2000}

>>> let m = mkDense (mat 2 [1..4])
>>> m
Dense {gmDense = (2><2)
 [ 1.0, 2.0
 , 3.0, 4.0 ], nRows = 2, nCols = 2}

Misc