Source file `owl_dense_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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# 1 "src/owl/dense/owl_dense_matrix_c.ml"
(*
 * OWL - OCaml Scientific and Engineering Computing
 * Copyright (c) 2016-2020 Liang Wang <liang.wang@cl.cam.ac.uk>
 *)

(** [ Complex dense matrix module ]  *)

open Bigarray
module M = Owl_dense_matrix_generic
include M

type elt = Complex.t

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

type cast_mat = (float, float32_elt) Owl_dense_matrix_generic.t

(* overload functions in Owl_dense_matrix_generic *)

let empty m n = M.empty Complex32 m n

let create m n a = M.create Complex32 m n a

let init m n f = M.init Complex32 m n f

let init_2d m n f = M.init_2d Complex32 m n f

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 sequential ?a ?step m n = M.sequential Complex32 ?a ?step m n

let uniform ?a ?b m n = M.uniform Complex32 ?a ?b m n

let gaussian ?mu ?sigma m n = M.gaussian Complex32 ?mu ?sigma m n

(* let semidef m = M.semidef Complex32 m *)

let linspace a b n = M.linspace Complex32 a b n

let logspace ?base a b n = M.logspace Complex32 ?base a b n

let meshgrid xa xb ya yb xn yn = M.meshgrid Complex32 xa xb ya yb xn yn

let bernoulli ?p d = M.bernoulli Complex32 ?p d

let unit_basis n i = M.unit_basis Complex32 n i

let hadamard n = M.hadamard Complex32 n

let magic n = M.magic Complex32 n

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

let of_arrays x = M.of_arrays Complex32 x

let load f = M.load Complex32 f

let load_txt ?sep f = M.load_txt Complex32 ?sep f

let load_npy f = M.load_npy Complex32 f

(* specific functions for complex64 matrix *)

let vector n = empty 1 n

let vector_ones n = ones 1 n

let vector_zeros n = zeros 1 n

let vector_uniform n = uniform 1 n

let re x = re_c2s x

let im x = im_c2s x

let complex = complex Float32 Complex32

let polar = polar Float32 Complex32

(* ends here *)