Module Algodiff.NN

val dropout : ?rate:float -> t -> t
val conv1d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val conv2d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val conv3d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val dilated_conv1d : ?padding:Owl_types.padding -> t -> t -> int array -> int array -> t
val dilated_conv2d : ?padding:Owl_types.padding -> t -> t -> int array -> int array -> t
val dilated_conv3d : ?padding:Owl_types.padding -> t -> t -> int array -> int array -> t
val transpose_conv1d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val transpose_conv2d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val transpose_conv3d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val max_pool1d : Owl_types.padding -> t -> int array -> int array -> t
val max_pool2d : Owl_types.padding -> t -> int array -> int array -> t
val max_pool3d : Owl_types.padding -> t -> int array -> int array -> t
val avg_pool1d : Owl_types.padding -> t -> int array -> int array -> t
val avg_pool2d : Owl_types.padding -> t -> int array -> int array -> t
val avg_pool3d : Owl_types.padding -> t -> int array -> int array -> t
val upsampling2d : t -> int array -> t
val pad : ?v:A.elt -> int list list -> t -> t