parameters method

@override List<Value> parameters()

Implementation

@override
List<Value> parameters() {
  return [
    ...pieceEmbeddings.expand((vec) => vec.values),
    ...squarePositionalEmbeddings.expand((vec) => vec.values),
    ...transformerEncoder.parameters(),
    ...moveHead.parameters(),
  ];
}