Trait burn_core::module::ModuleVisitor

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

Module visitor trait.

Required Methods

fn visit<const D: usize>(&mut self, id: &ParamId, tensor: &Tensor<B, D>)

Visit a tensor in the module.

Object Safety

This trait is not object safe.

Implementors