Trait burn::module::ModuleVisitor

pub trait ModuleVisitor<B>
where
    B: Backend,
{
    // Provided methods
    fn visit_float<const D: usize>(
        &mut self,
        _id: &ParamId,
        _tensor: &Tensor<B, D>,
    ) { ... }
    fn visit_int<const D: usize>(
        &mut self,
        _id: &ParamId,
        _tensor: &Tensor<B, D, Int>,
    ) { ... }
    fn visit_bool<const D: usize>(
        &mut self,
        _id: &ParamId,
        _tensor: &Tensor<B, D, Bool>,
    ) { ... }
}

Module visitor trait.

Provided Methods

fn visit_float<const D: usize>(&mut self, _id: &ParamId, _tensor: &Tensor<B, D>)

Visit a float tensor in the module.

fn visit_int<const D: usize>( &mut self, _id: &ParamId, _tensor: &Tensor<B, D, Int>, )

Visit an int tensor in the module.

fn visit_bool<const D: usize>( &mut self, _id: &ParamId, _tensor: &Tensor<B, D, Bool>, )

Visit a bool tensor in the module.

Object Safety

This trait is not object safe.

Implementors