Struct Reinitializer

pub struct Reinitializer<B>
where
    B: Backend,
{ /* private fields */ }

Overrides float and int tensors of burn modules.

This is useful for testing.

Implementations

impl<B> Reinitializer<B>

where B: Backend,

pub fn new() -> Reinitializer<B>

Create a new reinitializer.

pub fn apply<M>(self, module: M) -> M

where M: Module<B>,

Apply the reinitialization to the given module.

pub fn constant(self, constant: f64) -> Reinitializer<B>

Set the reinitialization strategy to constant for all tensors.

pub fn constant_float(self, constant: f64) -> Reinitializer<B>

Set the reinitialization strategy to constant for float tensors.

pub fn constant_int(self, constant: i64) -> Reinitializer<B>

Set the reinitialization strategy to constant for int tensors.

pub fn random(self, seed: u64, min: f64, max: f64) -> Reinitializer<B>

Set the reinitialization strategy to random for all tensors.

pub fn random_float(self, seed: u64, min: f64, max: f64) -> Reinitializer<B>

Set the reinitialization strategy to random for float tensors.

pub fn random_int(self, seed: u64, min: i64, max: i64) -> Reinitializer<B>

Set the reinitialization strategy to random for int tensors.

pub fn range(self, min: f64, max: f64) -> Reinitializer<B>

Set the reinitialization strategy to range for all tensors.

pub fn range_float(self, min: f64, max: f64) -> Reinitializer<B>

Set the reinitialization strategy to range for float tensors.

pub fn range_int(self, min: i64, max: i64) -> Reinitializer<B>

Set the reinitialization strategy to range for int tensors.

Trait Implementations

impl<B> Debug for Reinitializer<B>

where B: Debug + Backend,

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl<B> Default for Reinitializer<B>

where B: Backend,

fn default() -> Reinitializer<B>

Returns the “default value” for a type.

impl<B> ModuleMapper<B> for Reinitializer<B>

where B: Backend,

fn map_float<const D: usize>( &mut self, param: Param<Tensor<B, D>>, ) -> Param<Tensor<B, D>>

Map a float parameter in the module.

fn map_int<const D: usize>( &mut self, param: Param<Tensor<B, D, Int>>, ) -> Param<Tensor<B, D, Int>>

Map an int parameter in the module.

fn map_bool<const D: usize>( &mut self, param: Param<Tensor<B, D, Bool>>, ) -> Param<Tensor<B, D, Bool>>

Map a bool parameter in the module.

fn enter_module(&mut self, name: &str, container_type: &str)

Called when entering a submodule.

fn exit_module(&mut self, name: &str, container_type: &str)

Called when exiting a submodule.