Source file `owl_dense_matrix_z.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_z.ml"
(*
 * OWL - OCaml Scientific Computing
 * Copyright (c) 2016-2022 Liang Wang <liang@ocaml.xyz>
 *)

(** [ Complex dense matrix module ]  *)

open Bigarray
module M = Owl_dense_matrix_generic
include M

type elt = Complex.t

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

type cast_mat = (float, float64_elt) Owl_dense_matrix_generic.t

(* overload functions in Owl_dense_matrix_generic *)

let empty m n = M.empty Complex64 m n

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

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

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

let zeros m n = M.zeros Complex64 m n

let ones m n = M.ones Complex64 m n

let eye m = M.eye Complex64 m

let sequential ?a ?step m n = M.sequential Complex64 ?a ?step m n

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

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

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

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

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

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

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

let unit_basis n i = M.unit_basis Complex64 n i

let hadamard n = M.hadamard Complex64 n

let magic n = M.magic Complex64 n

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

let of_arrays x = M.of_arrays Complex64 x

let load f = M.load Complex64 f

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

let load_npy f = M.load_npy Complex64 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_z2d x

let im x = im_z2d x

let complex = complex Float64 Complex64

let polar = polar Float64 Complex64

(* ends here *)