tieTo method

void tieTo(
  1. Embedding emb
)

Implementation

void tieTo(Embedding emb) {
  if (emb.dim != inFeatures || emb.vocabSize != outFeatures) {
    throw ArgumentError('tieWeights: shape mismatch');
  }
  // Copy transpose: emb.weight: (vocab, dim) -> (dim, vocab)
  final t = _transpose(emb.weight);
  for (var i = 0; i < w.data.length; i++) {
    w.data[i] = t.data[i];
  }
}