parameters method

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

Trainable tensors owned by this module (and its submodules).

Implementation

@override
List<Tensor> parameters() => [
  ...down1.parameters(),
  ...down2.parameters(),
  ...mid.parameters(),
  ...timeProj.parameters(),
  ...up1.parameters(),
  ...up2.parameters(),
];