Trait burn_core::module::ModuleMapper

source · 
pub trait ModuleMapper<B: Backend> {
    // Required method
    fn map<const D: usize>(
        &mut self,
        id: &ParamId,
        tensor: Tensor<B, D>
    ) -> Tensor<B, D>;
}
Expand description

Module mapper trait.

Required Methods§

source

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

Map a tensor in the module.

Object Safety§

This trait is not object safe.

Implementors§