Source file owl_sparse_matrix_s.ml

# 1 "src/owl/sparse/owl_sparse_matrix_s.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.float32_elt) Owl_sparse_matrix_generic.t

(* overload functions in Owl_sparse_matrix_generic *)

let zeros m n = M.zeros Float32 m n

let ones m n = M.ones Float32 m n

let eye m = M.eye Float32 m

let binary m n = M.binary Float32 m n

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

let sequential m n = M.sequential Float32 m n

let permutation_matrix m = M.permutation_matrix Float32 m

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

let load f = M.load Float32 f

(* specific functions for float32 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 Float32 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 Float32 a b n |> of_dense

(** ends here *)