Module Ocannl.Nn_blocks

Prior to OCANNL 0.5, this module is just a placeholder hinting at an intended design pattern for model components.

module TDSL = Operation.TDSL
module NTDSL = Operation.NTDSL
type mlp_layer_config = {
  1. label : Base.string Base.list;
  2. hid_dim : Base.int;
}
val mlp_layer : config:mlp_layer_config -> Tensor.t -> Tensor.t
type mlp_config = {
  1. label : Base.string Base.list;
  2. hid_dims : Base.int Base.list;
}
val mlp : config:mlp_config -> Tensor.t -> Tensor.t