pub trait ModuleVisitor<B: Backend> { // Required method fn visit<const D: usize>(&mut self, id: &ParamId, tensor: &Tensor<B, D>); }