forked from tsai/mathnet-numerics
5 changed files with 261 additions and 9 deletions
@ -0,0 +1,198 @@ |
|||
// <copyright file="Matrix.fs" company="Math.NET"> |
|||
// Math.NET Numerics, part of the Math.NET Project |
|||
// http://mathnet.opensourcedotnet.info |
|||
// |
|||
// Copyright (c) 2009 Math.NET |
|||
// |
|||
// Permission is hereby granted, free of charge, to any person |
|||
// obtaining a copy of this software and associated documentation |
|||
// files (the "Software"), to deal in the Software without |
|||
// restriction, including without limitation the rights to use, |
|||
// copy, modify, merge, publish, distribute, sublicense, and/or sell |
|||
// copies of the Software, and to permit persons to whom the |
|||
// Software is furnished to do so, subject to the following |
|||
// conditions: |
|||
// |
|||
// The above copyright notice and this permission notice shall be |
|||
// included in all copies or substantial portions of the Software. |
|||
// |
|||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
|||
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES |
|||
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND |
|||
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT |
|||
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, |
|||
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
|||
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR |
|||
// OTHER DEALINGS IN THE SOFTWARE. |
|||
// </copyright> |
|||
|
|||
namespace MathNet.Numerics.LinearAlgebra.Double |
|||
|
|||
open MathNet.Numerics.LinearAlgebra.Double |
|||
|
|||
/// A module which implements functional matrix operations. |
|||
module Matrix = |
|||
|
|||
/// Fold a function over all matrix elements. |
|||
let inline fold (f: 'a -> float -> 'a) (acc0: 'a) (A: #Matrix) = |
|||
let n = A.RowCount |
|||
let m = A.ColumnCount |
|||
let mutable acc = acc0 |
|||
for i=0 to n-1 do |
|||
for j=0 to m-1 do |
|||
acc <- f acc (A.Item(i,j)) |
|||
acc |
|||
|
|||
/// Fold a matrix by applying a given function to all matrix elements. |
|||
let inline foldi (f: int -> int -> 'a -> float -> 'a) (acc0: 'a) (A: #Matrix) = |
|||
let n = A.RowCount |
|||
let m = A.ColumnCount |
|||
let mutable acc = acc0 |
|||
for i=0 to n-1 do |
|||
for j=0 to m-1 do |
|||
acc <- f i j acc (A.Item(i,j)) |
|||
acc |
|||
|
|||
/// Create a 2D array from a matrix. |
|||
let inline toArray2 (A: #Matrix) = |
|||
let n = A.RowCount |
|||
let m = A.ColumnCount |
|||
Array2D.init n m (fun i j -> (A.Item(i,j))) |
|||
|
|||
/// Checks whether a predicate holds for all elements of a matrix. |
|||
let inline forall (p: float -> bool) (A: #Matrix) = |
|||
let mutable b = true |
|||
let mutable i = 0 |
|||
let mutable j = 0 |
|||
while b && i < A.RowCount do |
|||
b <- b && (p (A.Item(i,j))) |
|||
j <- j+1 |
|||
if j = A.ColumnCount then i <- i+1; j <- 0 |
|||
b |
|||
|
|||
/// Chechks whether a predicate holds for at least one element of a matrix. |
|||
let inline exists (p: float -> bool) (A: #Matrix) = |
|||
let mutable b = false |
|||
let mutable i = 0 |
|||
let mutable j = 0 |
|||
while not(b) && i < A.RowCount do |
|||
b <- b || (p (A.Item(i,j))) |
|||
j <- j+1 |
|||
if j = A.ColumnCount then i <- i+1; j <- 0 |
|||
b |
|||
|
|||
/// Checks whether a position dependent predicate holds for all elements of a matrix. |
|||
let inline foralli (p: int -> int -> float -> bool) (A: #Matrix) = |
|||
let mutable b = true |
|||
let mutable i = 0 |
|||
let mutable j = 0 |
|||
while b && i < A.RowCount do |
|||
b <- b && (p i j (A.Item(i,j))) |
|||
j <- j+1 |
|||
if j = A.ColumnCount then i <- i+1; j <- 0 |
|||
b |
|||
|
|||
/// Checks whether a position dependent predicate holds for at least one element of a matrix. |
|||
let inline existsi (p: int -> int -> float -> bool) (A: #Matrix) = |
|||
let mutable b = false |
|||
let mutable i = 0 |
|||
let mutable j = 0 |
|||
while not(b) && i < A.RowCount do |
|||
b <- b || (p i j (A.Item(i,j))) |
|||
j <- j+1 |
|||
if j = A.ColumnCount then i <- i+1; j <- 0 |
|||
b |
|||
|
|||
/// Map every matrix element using the given function. |
|||
let inline map (f: float -> float) (A: #Matrix) = |
|||
let N = A.RowCount |
|||
let M = A.ColumnCount |
|||
let C = A.Clone() |
|||
for i=0 to N-1 do |
|||
for j=0 to M-1 do |
|||
C.[i,j] <- f (C.Item(i,j)) |
|||
C |
|||
|
|||
/// Map every matrix element using the given position dependent function. |
|||
let inline mapi (f: int -> int -> float -> float) (A: #Matrix) = |
|||
let N = A.RowCount |
|||
let M = A.ColumnCount |
|||
let C = A.Clone() |
|||
for i=0 to N-1 do |
|||
for j=0 to M-1 do |
|||
C.[i,j] <- f i j (C.Item(i,j)) |
|||
C |
|||
|
|||
/// In-place assignment. |
|||
let inline inplaceAssign (f: int -> int -> float) (A: #Matrix) = |
|||
for i=0 to A.RowCount-1 do |
|||
for j=0 to A.ColumnCount-1 do |
|||
A.Item(i,j) <- f i j |
|||
|
|||
/// In-place map of every matrix element using a position dependent function. |
|||
let inline inplaceMapi (f: int -> int -> float -> float) (A: #Matrix) = |
|||
for i=0 to A.RowCount-1 do |
|||
for j=0 to A.ColumnCount-1 do |
|||
A.Item(i,j) <- f i j (A.Item(i,j)) |
|||
|
|||
/// Creates a sequence that iterates the non-zero entries in the matrix. |
|||
let inline nonZeroEntries (A: #Matrix) = |
|||
seq { for i in 0 .. A.RowCount-1 do |
|||
for j in 0 .. A.ColumnCount-1 do |
|||
if A.Item(i,j) <> 0.0 then yield (i,j, A.Item(i,j)) } |
|||
|
|||
/// Returns the sum of all elements of a matrix. |
|||
let inline sum (A: #Matrix) = |
|||
let mutable f = 0.0 |
|||
for i=0 to A.RowCount-1 do |
|||
for j=0 to A.ColumnCount-1 do |
|||
f <- f + A.Item(i,j) |
|||
f |
|||
|
|||
/// Iterates over all elements of a matrix. |
|||
let inline iter (f: float -> unit) (A: #Matrix) = |
|||
for i=0 to A.RowCount-1 do |
|||
for j=0 to A.ColumnCount-1 do |
|||
f (A.Item(i,j)) |
|||
() |
|||
|
|||
/// Iterates over all elements of a matrix using the element indices. |
|||
let inline iteri (f: int -> int -> float -> unit) (A: #Matrix) = |
|||
for i=0 to A.RowCount-1 do |
|||
for j=0 to A.ColumnCount-1 do |
|||
f i j (A.Item(i,j)) |
|||
() |
|||
|
|||
/// Fold one column. |
|||
let inline foldCol (f: 'a -> float -> 'a) acc (A: #Matrix) k = |
|||
let mutable macc = acc |
|||
for i=0 to A.RowCount-1 do |
|||
macc <- f macc (A.Item(i,k)) |
|||
macc |
|||
|
|||
/// Fold one row. |
|||
let inline foldRow (f: 'a -> float -> 'a) acc (A: #Matrix) k = |
|||
let mutable macc = acc |
|||
for i=0 to A.ColumnCount-1 do |
|||
macc <- f macc (A.Item(k,i)) |
|||
macc |
|||
|
|||
/// Fold all columns into one row vector. |
|||
let inline foldByCol (f: float -> float -> float) acc (A: #Matrix) = |
|||
let v = new DenseVector(A.ColumnCount) |
|||
for k=0 to A.ColumnCount-1 do |
|||
let mutable macc = acc |
|||
for i=0 to A.RowCount-1 do |
|||
macc <- f macc (A.Item(i,k)) |
|||
v.[k] <- macc |
|||
v :> Vector |
|||
|
|||
/// Fold all rows into one column vector. |
|||
let inline foldByRow (f: float -> float -> float) acc (A: #Matrix) = |
|||
let v = new DenseVector(A.RowCount) |
|||
for k=0 to A.RowCount-1 do |
|||
let mutable macc = acc |
|||
for i=0 to A.ColumnCount-1 do |
|||
macc <- f macc (A.Item(k,i)) |
|||
v.[k] <- macc |
|||
v :> Vector |
|||
Loading…
Reference in new issue