MLP constructor

MLP(
  1. int nin,
  2. List<int> nouts
)

Implementation

MLP(int nin, List<int> nouts) {
  List<int> dims = [nin, ...nouts];
  for (int i = 0; i < nouts.length; i++) {
    layers.add(Layer(dims[i], dims[i + 1], useGelu: i != nouts.length - 1));
  }
}