|
| 1 | +// Learn more about F# at http://fsharp.org |
| 2 | +// See the 'F# Tutorial' project for more help. |
| 3 | + |
| 4 | +open System |
| 5 | +open System.Numerics |
| 6 | +open System.Collections.Generic |
| 7 | +open ArrayFire |
| 8 | + |
| 9 | +[<EntryPoint>] |
| 10 | +let main argv = |
| 11 | + |
| 12 | + Device.SetBackend(Backend.DEFAULT) |
| 13 | + Device.PrintInfo() |
| 14 | + printfn "\n=== ArrayFire signed(s32) / unsigned(u32) Integer Example ===\n" |
| 15 | + |
| 16 | + let h_A = array2D[ [ 1; 2; 4 ]; [ -1; 2; 0 ]; [ 4; 2; 3 ] ] |
| 17 | + let h_B = array2D[ [ 2; 3; -5 ]; [ 6; 0; 10 ]; [ -12; 0; 1 ] ] |
| 18 | + //printfn "Array 2D is %A" h_A.GetType() |
| 19 | + let A = Data.CreateArray(h_A) |
| 20 | + let B = Data.CreateArray(h_B) |
| 21 | + printfn "--\nSub-refencing and Sub-assignment\n" |
| 22 | + |
| 23 | + Util.Print(A, "A"); |
| 24 | + Util.Print(A.Rows(0, 0), "A's first row") |
| 25 | + Util.Print(A.Cols(0, 0), "A's first column") |
| 26 | + A.[Util.Seq(0, 0), Util.Seq(0, 0)] <- Data.CreateArray([|100|]) |
| 27 | + A.[Util.Seq(1, 1), Util.Seq(2, 2)] <- Data.CreateArray([|100|]) |
| 28 | + Util.Print(A, "A") |
| 29 | + Util.Print(B, "B") |
| 30 | + A.[Util.Seq(1, 1), Util.Span] <- B.[Util.Seq(2, 2), Util.Span] |
| 31 | + Util.Print(A, "A") |
| 32 | + Util.Print(B, "B") |
| 33 | + let C = A + B |
| 34 | + printf "--Bit-wise operations\n" |
| 35 | + Util.Print(A &&& B, "A &&& B") |
| 36 | + Util.Print(A ||| B, "A ||| B") |
| 37 | + Util.Print(A ^^^ B, "A ^^^ B") |
| 38 | + |
| 39 | + printfn "\n--Transpose\n" |
| 40 | + Util.Print(A, "A") |
| 41 | + Util.Print(Matrix.Transpose(A, false), "Matrix.Transpose(A)") |
| 42 | + |
| 43 | + printfn "\n--Sum along columns\n" |
| 44 | + Util.Print(A, "A") |
| 45 | + Util.Print(Algorithm.Sum(A), "Algorithm.Sum(A)") |
| 46 | + |
| 47 | + printfn "\n--Product along columns\n" |
| 48 | + Util.Print(A, "A") |
| 49 | + Util.Print(Algorithm.Product(A), "Algorithm.Product(A)") |
| 50 | + |
| 51 | + printfn "\n--Minimum along columns\n" |
| 52 | + Util.Print(A, "A") |
| 53 | + Util.Print(Algorithm.Min(A), "Algorithm.Min(A)") |
| 54 | + |
| 55 | + printfn "\n--Maximum along columns\n" |
| 56 | + Util.Print(A, "A") |
| 57 | + Util.Print(Algorithm.Max(A), "Algorithm.Max(A)") |
| 58 | + |
| 59 | + printfn "\n--Minimum along columns with index\n" |
| 60 | + Util.Print(A, "A") |
| 61 | + |
| 62 | + let mutable idx = null |
| 63 | + let outArray = Algorithm.Min(A, &idx) |
| 64 | + Util.Print(outArray, "output") |
| 65 | + Util.Print(idx, "indexes") |
| 66 | + |
| 67 | + 0 // return an integer exit code |
0 commit comments