package owl

  1. Overview
  2. Docs
package owl
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

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

(** [ Sparse matrix ] *)

open Bigarray
module M = Owl_sparse_matrix_generic
include M

type elt = float

type mat = (float, Bigarray.float64_elt) Owl_sparse_matrix_generic.t

(* overload functions in Owl_sparse_matrix_generic *)

let zeros m n = M.zeros Float64 m n

let ones m n = M.ones Float64 m n

let eye m = M.eye Float64 m

let binary m n = M.binary Float64 m n

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

let sequential m n = M.sequential Float64 m n

let permutation_matrix m = M.permutation_matrix Float64 m

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

let load f = M.load Float64 f

(* specific functions for float64 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 Float64 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 = 0) ?(b = 99) m n =
  _random_basic (fun () -> float_of_int (Owl_stats.uniform_int_rvs ~a ~b)) m n


let linspace a b n = Owl_dense_matrix_generic.linspace Float64 a b n |> of_dense

(** ends here *)
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct