parameters method

@override List<Tensor> parameters()

Implementation

@override
List<Tensor> parameters() => [
  wte,
  wpe,
  ...blocks.expand((block) => block.parameters()),
  ...finalLayerNorm.parameters(),
];