Trait burn::module::ModuleMapper

pub trait ModuleMapper<B>
where
    B: Backend,
{
    // Provided methods
    fn map_float<const D: usize>(
        &mut self,
        _id: &ParamId,
        tensor: Tensor<B, D>
    ) -> Tensor<B, D> { ... }
    fn map_int<const D: usize>(
        &mut self,
        _id: &ParamId,
        tensor: Tensor<B, D, Int>
    ) -> Tensor<B, D, Int> { ... }
    fn map_bool<const D: usize>(
        &mut self,
        _id: &ParamId,
        tensor: Tensor<B, D, Bool>
    ) -> Tensor<B, D, Bool> { ... }
}

Module mapper trait.

Provided Methods

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

Map a float tensor in the module.

fn map_int<const D: usize>( &mut self, _id: &ParamId, tensor: Tensor<B, D, Int> ) -> Tensor<B, D, Int>

Map an int tensor in the module.

fn map_bool<const D: usize>( &mut self, _id: &ParamId, tensor: Tensor<B, D, Bool> ) -> Tensor<B, D, Bool>

Map a bool tensor in the module.

Object Safety

This trait is not object safe.

Implementors