package eigen

  1. Overview
  2. Docs
Owl's OCaml interface to Eigen3 C++ library

Install 

dune-project
 Dependency

github.com Readme Edit opam file Versions (14)

Authors

  1. L
    Liang Wang

Maintainers

  1. L
    Liang Wang <ryanrhymes@gmail.com>

Sources

eigen-0.3.1.tbz
sha256=dcb882701d1266550489f1c08f24dd14f840f1beb6f730fe84e87c9ed9a602e4
sha512=3efbf403d3daa765d7a5e6695c276566f9ff72c5e266044c6ae19d168c86dd87ddc9eb8bfa75417837cc340944b30e798e6f5d2417a0f53c48316dae9149ae7b

doc/src/eigen/eigen_dsmat_s.ml.html

Source file eigen_dsmat_s.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
(*
 * Eigen - an OCaml interface to C++ Eigen library
 * Copyright (c) 2016-2020 Liang Wang <liang.wang@cl.cam.ac.uk>
 *)

open Ctypes
open Eigen_types.DSMAT_S

let create m n =
  let x = ml_eigen_new (Int64.of_int m) (Int64.of_int n) in
  Gc.finalise ml_eigen_delete x;
  x

let delete x = ml_eigen_delete x

let zeros m n =
  let x = ml_eigen_zeros (Int64.of_int m) (Int64.of_int n) in
  Gc.finalise ml_eigen_delete x;
  x

let ones m n =
  let x = ml_eigen_ones (Int64.of_int m) (Int64.of_int n) in
  Gc.finalise ml_eigen_delete x;
  x

let eye m =
  let x = ml_eigen_eye (Int64.of_int m) in
  Gc.finalise ml_eigen_delete x;
  x

let rows x = ml_eigen_rows x |> Int64.to_int

let cols x = ml_eigen_cols x |> Int64.to_int

let get x i j = ml_eigen_get x (Int64.of_int i) (Int64.of_int j)

let set x i j a = ml_eigen_set x (Int64.of_int i) (Int64.of_int j) a

let data x = ml_eigen_data x

let to_bigarray x = bigarray_of_ptr array2 (rows x, cols x) Bigarray.float32 (data x)

let of_bigarray x =
  let _ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let m = Bigarray.Array2.dim1 x |> Int64.of_int in
  let n = Bigarray.Array2.dim2 x |> Int64.of_int in
  ml_eigen_of_bigarray _ptr m n

let of_bigarray2 x =
  let _mat = Ctypes.allocate (ptr void) null |> Obj.repr |> Obj.obj in
  let _ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let m = Bigarray.Array2.dim1 x |> Int64.of_int in
  let n = Bigarray.Array2.dim2 x |> Int64.of_int in
  ml_eigen_of_bigarray2 _mat _ptr m n;
  _mat

let print x = ml_eigen_print x

let gemm x y =
  let x_ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let xm = Bigarray.Array2.dim1 x |> Int64.of_int in
  let xn = Bigarray.Array2.dim2 x |> Int64.of_int in
  let y_ptr = Ctypes.bigarray_start Ctypes_static.Array2 y in
  let ym = Bigarray.Array2.dim1 y |> Int64.of_int in
  let yn = Bigarray.Array2.dim2 y |> Int64.of_int in
  let z = ml_eigen_gemm x_ptr xm xn y_ptr ym yn in
  Gc.finalise ml_eigen_delete z;
  to_bigarray z

let transpose x =
  let x_ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let m = Bigarray.Array2.dim1 x in
  let n = Bigarray.Array2.dim2 x in
  let xm = Int64.of_int m in
  let xn = Int64.of_int n in
  let y = ml_eigen_transpose x_ptr xm xn in
  Gc.finalise ml_eigen_delete y;
  to_bigarray y

let swap_rows x i j =
  let ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let m = Bigarray.Array2.dim1 x |> Int64.of_int in
  let n = Bigarray.Array2.dim2 x |> Int64.of_int in
  let i = Int64.of_int i in
  let j = Int64.of_int j in
  let _ = ml_eigen_swap_rows ptr m n i j in
  ()

let swap_cols x i j =
  let ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let m = Bigarray.Array2.dim1 x |> Int64.of_int in
  let n = Bigarray.Array2.dim2 x |> Int64.of_int in
  let i = Int64.of_int i in
  let j = Int64.of_int j in
  let _ = ml_eigen_swap_cols ptr m n i j in
  ()

let rowwise_op op x y =
  let x_ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let y_ptr = Ctypes.bigarray_start Ctypes_static.Array2 y in
  let m = Bigarray.Array2.dim1 x |> Int64.of_int in
  let n = Bigarray.Array2.dim2 x |> Int64.of_int in
  let _ = ml_eigen_rowwise_op op x_ptr m n y_ptr in
  ()

let colwise_op op x y =
  let x_ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let y_ptr = Ctypes.bigarray_start Ctypes_static.Array2 y in
  let m = Bigarray.Array2.dim1 x |> Int64.of_int in
  let n = Bigarray.Array2.dim2 x |> Int64.of_int in
  let _ = ml_eigen_colwise_op op x_ptr m n y_ptr in
  ()

let inv x =
  let x_ptr = Ctypes.bigarray_start Ctypes_static.Array2 x in
  let xm = Bigarray.Array2.dim1 x |> Int64.of_int in
  let xn = Bigarray.Array2.dim2 x |> Int64.of_int in
  let z = ml_eigen_inv x_ptr xm xn in
  Gc.finalise ml_eigen_delete z;
  to_bigarray z