package nx

  1. Overview
  2. Docs
N-dimensional arrays for OCaml

Install 

dune-project
 Dependency

github.com Readme Changelog Edit opam file Versions (4)

Authors

  1. Thibaut Mattio

Maintainers

  1. Thibaut Mattio

Sources

raven-1.0.0.alpha3.tbz
sha256=96d35ce03dfbebd2313657273e24c2e2d20f9e6c7825b8518b69bd1d6ed5870f
sha512=90c5053731d4108f37c19430e45456063e872b04b8a1bbad064c356e1b18e69222de8bfcf4ec14757e71f18164ec6e4630ba770dbcb1291665de5418827d1465

doc/src/nx.pocketfft/pocketfft.ml.html

Source file pocketfft.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
(** PocketFFT bindings *)

external c2c_f32 :
     shape:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> forward:bool
  -> fct:float
  -> data_in:(Complex.t, Bigarray.complex32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(Complex.t, Bigarray.complex32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_c2c_f32_bytecode" "caml_pocketfft_c2c_f32"
[@@noalloc]

external r2c_f32 :
     shape_in:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> forward:bool
  -> fct:float
  -> data_in:(float, Bigarray.float32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(Complex.t, Bigarray.complex32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_r2c_f32_bytecode" "caml_pocketfft_r2c_f32"
[@@noalloc]

external c2r_f32 :
     shape_out:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> forward:bool
  -> fct:float
  -> data_in:(Complex.t, Bigarray.complex32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(float, Bigarray.float32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_c2r_f32_bytecode" "caml_pocketfft_c2r_f32"
[@@noalloc]

external dct_f32 :
     shape:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> dct_type:int
  -> ortho:bool
  -> fct:float
  -> data_in:(float, Bigarray.float32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(float, Bigarray.float32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_dct_f32_bytecode" "caml_pocketfft_dct_f32"
[@@noalloc]

external dst_f32 :
     shape:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> dct_type:int
  -> ortho:bool
  -> fct:float
  -> data_in:(float, Bigarray.float32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(float, Bigarray.float32_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_dst_f32_bytecode" "caml_pocketfft_dst_f32"
[@@noalloc]

external c2c_f64 :
     shape:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> forward:bool
  -> fct:float
  -> data_in:(Complex.t, Bigarray.complex64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(Complex.t, Bigarray.complex64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_c2c_f64_bytecode" "caml_pocketfft_c2c_f64"
[@@noalloc]

external r2c_f64 :
     shape_in:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> forward:bool
  -> fct:float
  -> data_in:(float, Bigarray.float64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(Complex.t, Bigarray.complex64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_r2c_f64_bytecode" "caml_pocketfft_r2c_f64"
[@@noalloc]

external c2r_f64 :
     shape_out:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> forward:bool
  -> fct:float
  -> data_in:(Complex.t, Bigarray.complex64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(float, Bigarray.float64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_c2r_f64_bytecode" "caml_pocketfft_c2r_f64"
[@@noalloc]

external dct_f64 :
     shape:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> dct_type:int
  -> ortho:bool
  -> fct:float
  -> data_in:(float, Bigarray.float64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(float, Bigarray.float64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_dct_f64_bytecode" "caml_pocketfft_dct_f64"
[@@noalloc]

external dst_f64 :
     shape:int array
  -> stride_in:int array
  -> stride_out:int array
  -> axes:int array
  -> dct_type:int
  -> ortho:bool
  -> fct:float
  -> data_in:(float, Bigarray.float64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> data_out:(float, Bigarray.float64_elt, Bigarray.c_layout) Bigarray.Array1.t
  -> nthreads:int
  -> unit = "caml_pocketfft_dst_f64_bytecode" "caml_pocketfft_dst_f64"
[@@noalloc]