owl-base 0.3.8 · OCaml Package

Helper functions used in the library

include module type of struct include Owl_utils_conv end

val calc_conv2d_output_shape : 
  Owl_types.padding ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int * int

val calc_transpose_conv2d_output_shape : 
  Owl_types.padding ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int * int

val calc_conv2d_padding : 
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int * int * int * int

val calc_conv1d_output_shape : Owl_types.padding -> int -> int -> int -> int

val calc_transpose_conv1d_output_shape : 
  Owl_types.padding ->
  int ->
  int ->
  int ->
  int

val calc_conv3d_output_shape : 
  Owl_types.padding ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int * int * int

val calc_transpose_conv3d_output_shape : 
  Owl_types.padding ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int * int * int

val calc_conv3d_padding : 
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int ->
  int * int * int * int * int * int

include module type of struct include Owl_utils_ndarray end

val elt_to_str : 'a 'b. ('a, 'b) Bigarray.kind -> 'a -> string

val elt_of_str : 'a 'b. ('a, 'b) Bigarray.kind -> string -> 'a

val numel : ('a, 'b, 'c) Bigarray.Genarray.t -> int

val calc_stride : int array -> int array

val calc_slice : int array -> int array

val index_1d_nd : int -> int array -> int array -> unit

val index_nd_1d : int array -> int array -> int

val ind : ('a, 'b, 'c) Bigarray.Genarray.t -> int -> int array

val i1d : ('a, 'b, 'c) Bigarray.Genarray.t -> int array -> int

val adjust_index : int -> int -> int

val reduce_params : 
  int ->
  ('a, 'b, 'c) Bigarray.Genarray.t ->
  int * int * int * int array

module Stack = Owl_utils_stack

module Array = Owl_utils_array

val range_fold : int -> int -> f:('a -> int -> 'a) -> init:'a -> 'a

val array_reverse : 'a array -> unit

val array_insert : 'a array -> int -> 'a -> 'a array

val get_suffix : string -> string

val count_dup : 'a list -> ('a * int) list

val list_search : 'a -> 'a list -> int

val array2_to_array1 : 
  ('a, 'b, 'c) Bigarray.Array2.t ->
  ('a, 'b, 'c) Bigarray.Array1.t

val aarr_iter : ('a -> unit) -> 'a array array -> unit

val aarr_iteri : (int -> int -> 'a -> unit) -> 'a array array -> unit

val aarr_map : ('a -> 'b) -> 'a array array -> 'b array array

val aarr_mapi : (int -> int -> 'a -> 'b) -> 'a array array -> 'b array array

val aarr_map2 : 
  ('a -> 'b -> 'c) ->
  'a array array ->
  'b array array ->
  'c array array

val aarr_map2i : 
  (int -> int -> 'a -> 'b -> 'c) ->
  'a array array ->
  'b array array ->
  'c array array

val aarr_map3i : 
  (int -> int -> 'a -> 'b -> 'c -> 'd) ->
  'a array array ->
  'b array array ->
  'c array array ->
  'd array array

val aarr_map3 : 
  ('a -> 'b -> 'c -> 'd) ->
  'a array array ->
  'b array array ->
  'c array array ->
  'd array array

val aarr_map4 : 
  ('a -> 'b -> 'c -> 'd -> 'e) ->
  'a array array ->
  'b array array ->
  'c array array ->
  'd array array ->
  'e array array

val aarr2llss : 'a array array -> 'a list list

val llss2aarr : 'a list list -> 'a array array

val aarr_fold : ('a -> 'b -> 'a) -> 'a -> 'b array array -> 'a

val aarr_matrix : int -> int -> 'a -> 'a array array

val array1_extend : 
  ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t ->
  int ->
  ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t

val array1_copy : 
  ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t ->
  ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t

val format_time : float -> string

val time : (unit -> 'a) -> float

measure the time spent in a function in millisecond

val eps : 'a 'b. ('a, 'b) Bigarray.kind -> float

TODO: return the the distance between 1.0 and the next larger representable floating-point value.

package owl-base