package neural_nets_lib

  1. Overview
  2. Docs

Module Ocannl.Nn_blocksSource

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

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