Legend:
Library
Module
Module type
Parameter
Class
Class type
Library
Module
Module type
Parameter
Class
Class type
include Owl_types_ndarray_eltcmp.Sig with type elt = float
include Owl_types_ndarray_basic.Sig with type elt = float
val number : Owl_types_common.number
val empty : int array -> arr
val zeros : int array -> arr
val ones : int array -> arr
val shape : arr -> int array
val numel : arr -> int
val get_fancy : Owl_types_common.index list -> arr -> arr
val set_fancy : Owl_types_common.index list -> arr -> arr -> unit
val reset : arr -> unit
val conv1d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val conv2d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val conv3d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val dilated_conv1d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
int array ->
arr
val dilated_conv2d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
int array ->
arr
val dilated_conv3d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
int array ->
arr
val transpose_conv1d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val transpose_conv2d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val transpose_conv3d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val max_pool1d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val max_pool2d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val max_pool3d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val avg_pool1d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val avg_pool2d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val avg_pool3d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val max_pool1d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val max_pool2d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val max_pool3d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val avg_pool1d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val avg_pool2d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val avg_pool3d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val row_num : arr -> int
val col_num : arr -> int
val float_to_elt : float -> elt
val elt_to_float : elt -> float
module Scalar : sig ... end
module Mat : sig ... end
module Linalg : sig ... end