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

open Bigarray
module M = Owl_base_dense_ndarray_generic
include M

type elt = Complex.t

type arr = (Complex.t, complex32_elt, c_layout) Genarray.t

let number = Owl_types.C32

let empty dims = M.empty Complex32 dims

let create dims value = M.create Complex32 dims value

let init dims f = M.init Complex32 dims f

let init_nd dims f = M.init_nd Complex32 dims f

let zeros dims = M.zeros Complex32 dims

let ones dims = M.ones Complex32 dims

let sequential ?a ?step dims = M.sequential Complex32 ?a ?step dims

let of_array arr dims = M.of_array Complex32 arr dims

let of_arrays arrays = M.of_arrays Complex32 arrays

let uniform ?a ?b dims = M.uniform Complex32 ?a ?b dims

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

let gaussian ?mu ?sigma dims = M.gaussian Complex32 ?mu ?sigma dims