Layer constructor

Layer(
  1. int nIn,
  2. int nOut, {
  3. bool useGelu = true,
})

Implementation

Layer(int nIn, int nOut, {this.useGelu = true})
  : weights = Tensor.xavier([nIn, nOut]),
    bias = Tensor.zeros([1, nOut]);