parameters method

  1. @override
List<Tensor> parameters()
override

Implementation

@override
List<Tensor> parameters() => [
  ...selfAttention.parameters(),
  ...crossAttention.parameters(),
  ...ffn.parameters(),
  ...ln1.parameters(),
  ...ln2.parameters(),
  ...ln3.parameters(),
];