Embedding constructor

Embedding(
  1. int vocabSize,
  2. int dim, {
  3. Tensor2D? weight,
  4. int seed = 7,
})

Implementation

Embedding(this.vocabSize, this.dim, {Tensor2D? weight, int seed = 7})
    : weight = weight ?? _xavier(vocabSize, dim, seed: seed) {
  assert(vocabSize > 0 && dim > 0);
}