burn::module

Trait ModuleVisitor

Source 
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>,
    ) { ... }
}
Expand description

Module visitor trait.

Provided Methods§

Source

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

Visit a float tensor in the module.

Source

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

Visit an int tensor in the module.

Source

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

Visit a bool tensor in the module.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§