Source file `owl_sparse_matrix_c.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
# 1 "src/owl/sparse/owl_sparse_matrix_c.ml"
(*
 * OWL - OCaml Scientific and Engineering Computing
 * Copyright (c) 2016-2020 Liang Wang <liang.wang@cl.cam.ac.uk>
 *)

(** [ Complex sparse matrix ]
  The default format is compressed row storage (CRS).
 *)

open Bigarray
module M = Owl_sparse_matrix_generic
include M

type elt = Complex.t

type mat = (Complex.t, Bigarray.complex32_elt) Owl_sparse_matrix_generic.t

(* overload functions in Owl_sparse_matrix *)

let zeros m n = M.zeros Complex32 m n

let ones m n = M.ones Complex32 m n

let eye m = M.eye Complex32 m

let binary m n = M.binary Complex32 m n

let uniform ?(scale = 1.) m n = M.uniform ~scale Complex32 m n

let sequential m n = M.sequential Complex32 m n

let permutation_matrix m = M.permutation_matrix Complex32 m

let of_array m n x = M.of_array Complex32 m n x

let load f = M.load Complex32 f

(* specific functions for complex32 matrix *)

let _random_basic f m n =
  let c = int_of_float (float_of_int (m * n) *. 0.15) in
  let x = M.zeros ~density:0.2 Complex32 m n in
  let l = Owl_stats.choose (Array.init (m * n) (fun i -> i)) c in
  for k = 0 to c - 1 do
    let i = l.(k) / n in
    let j = l.(k) - (i * n) in
    insert x i j (f ())
  done;
  x


let uniform_int ?(a = 1) ?(b = 99) m n =
  _random_basic
    (fun () ->
      let re = Owl_stats.uniform_int_rvs ~a ~b |> float_of_int in
      let im = Owl_stats.uniform_int_rvs ~a ~b |> float_of_int in
      Complex.{ re; im })
    m
    n


(** ends here *)