LinearLMHead constructor

LinearLMHead(
  1. int inFeatures,
  2. int outFeatures
)

Implementation

LinearLMHead(this.inFeatures, this.outFeatures)
    : w = Tensor2D.zeros(inFeatures, outFeatures),
      b = Float64List(outFeatures);