Source file `owl_base_algodiff_primal_ops.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
# 1 "src/base/algodiff/owl_base_algodiff_primal_ops.ml"
(*
 * OWL - OCaml Scientific Computing
 * Copyright (c) 2016-2022 Liang Wang <liang@ocaml.xyz>
 *)

module S = struct
  include Owl_base_dense_ndarray.S
  module Scalar = Owl_base_maths

  module Mat = struct
    let eye = Owl_base_dense_matrix_s.eye

    let tril = Owl_base_dense_matrix_s.tril

    let triu = Owl_base_dense_matrix_s.triu

    let diagm = Owl_base_dense_matrix_s.diagm
  end

  module Linalg = struct
    include Owl_base_linalg_s

    let qr a =
      let q, r, _ = qr a in
      q, r


    let lq x = lq x
  end
end

module D = struct
  include Owl_base_dense_ndarray_d
  module Scalar = Owl_base_maths

  module Mat = struct
    let eye = Owl_base_dense_matrix_d.eye

    let tril = Owl_base_dense_matrix_d.tril

    let triu = Owl_base_dense_matrix_d.triu

    let diagm = Owl_base_dense_matrix_d.diagm
  end

  module Linalg = struct
    include Owl_base_linalg_d

    let qr a =
      let q, r, _ = qr a in
      q, r


    let lq x = lq x
  end
end