Linear constructor

Linear(
  1. int inFeatures,
  2. int outFeatures, {
  3. bool useBias = true,
})

Implementation

Linear(this.inFeatures, this.outFeatures, {bool useBias = true}) {
  weight = Tensor.zeros([outFeatures, inFeatures]);
  if (useBias) {
    bias = Tensor.zeros([outFeatures]);
  }
}