Closed OliverFM closed 1 year ago
Further progress: this makes Linear an nn::Module, and adds a new nn::Module which is a sequence of nn::Linear modules. I would love to use const generics to check that the shapes are correct statically, but I think that this may be tricky.
Linear
nn::Module
nn::Linear
Further progress: this makes
Linearannn::Module, and adds a newnn::Modulewhich is a sequence ofnn::Linearmodules. I would love to use const generics to check that the shapes are correct statically, but I think that this may be tricky.