NN (owl-base.Owl_algodiff

val dropout : ?rate:float -> Core.t -> Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val conv1d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val conv2d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val conv3d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val dilated_conv1d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val dilated_conv2d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val dilated_conv3d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val transpose_conv1d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val transpose_conv2d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val transpose_conv3d : 
  ?padding:Owl_types.padding ->
  Core.t ->
  Core.t ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val max_pool1d : 
  Owl_types.padding ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val max_pool2d : 
  Owl_types.padding ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val max_pool3d : 
  Owl_types.padding ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val avg_pool1d : 
  Owl_types.padding ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val avg_pool2d : 
  Owl_types.padding ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val avg_pool3d : 
  Owl_types.padding ->
  Core.t ->
  int array ->
  int array ->
  Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val upsampling2d : Core.t -> int array -> Core.t

Refer to :doc:`owl_dense_ndarray_generic`

val pad : ?v:Core.A.elt -> int list list -> Core.t -> Core.t

Refer to :doc:`owl_dense_ndarray_generic`