diff options
Diffstat (limited to 'lib/Data/Packed/Matrix.hs')
-rw-r--r-- | lib/Data/Packed/Matrix.hs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/Data/Packed/Matrix.hs b/lib/Data/Packed/Matrix.hs index 2e8cb3d..45aaaba 100644 --- a/lib/Data/Packed/Matrix.hs +++ b/lib/Data/Packed/Matrix.hs | |||
@@ -77,7 +77,7 @@ diagRect s r c | |||
77 | | r > c = joinVert [diag s , zeros (r-c,c)] | 77 | | r > c = joinVert [diag s , zeros (r-c,c)] |
78 | where zeros (r,c) = reshape c $ constant 0 (r*c) | 78 | where zeros (r,c) = reshape c $ constant 0 (r*c) |
79 | 79 | ||
80 | takeDiag :: (Storable t) => Matrix t -> Vector t | 80 | takeDiag :: (Field t) => Matrix t -> Vector t |
81 | takeDiag m = fromList [cdat m `at` (k*cols m+k) | k <- [0 .. min (rows m) (cols m) -1]] | 81 | takeDiag m = fromList [cdat m `at` (k*cols m+k) | k <- [0 .. min (rows m) (cols m) -1]] |
82 | 82 | ||
83 | ident :: (Num t, Field t) => Int -> Matrix t | 83 | ident :: (Num t, Field t) => Int -> Matrix t |
@@ -119,7 +119,7 @@ dropColumns n mat = subMatrix (0,n) (rows mat, cols mat - n) mat | |||
119 | @\> flatten ('ident' 3) | 119 | @\> flatten ('ident' 3) |
120 | 9 # [1.0,0.0,0.0,0.0,1.0,0.0,0.0,0.0,1.0]@ | 120 | 9 # [1.0,0.0,0.0,0.0,1.0,0.0,0.0,0.0,1.0]@ |
121 | -} | 121 | -} |
122 | flatten :: Matrix t -> Vector t | 122 | flatten :: Field t => Matrix t -> Vector t |
123 | flatten = cdat | 123 | flatten = cdat |
124 | 124 | ||
125 | -- | Creates a 'Matrix' from a list of lists (considered as rows). | 125 | -- | Creates a 'Matrix' from a list of lists (considered as rows). |