Struct Reinitializer 

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

Available on crate feature std only.

Overrides float and int tensors of burn modules.

This is useful for testing.

Implementations

impl<B: Backend> Reinitializer<B>

pub fn new() -> Self

Create a new reinitializer.

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

Apply the reinitialization to the given module.

pub fn constant(self, constant: f64) -> Self

Set the reinitialization strategy to constant for all tensors.

pub fn constant_float(self, constant: f64) -> Self

Set the reinitialization strategy to constant for float tensors.

pub fn constant_int(self, constant: i64) -> Self

Set the reinitialization strategy to constant for int tensors.

pub fn random(self, seed: u64, min: f64, max: f64) -> Self

Set the reinitialization strategy to random for all tensors.

pub fn random_float(self, seed: u64, min: f64, max: f64) -> Self

Set the reinitialization strategy to random for float tensors.

pub fn random_int(self, seed: u64, min: i64, max: i64) -> Self

Set the reinitialization strategy to random for int tensors.

pub fn range(self, min: f64, max: f64) -> Self

Set the reinitialization strategy to range for all tensors.

pub fn range_float(self, min: f64, max: f64) -> Self

Set the reinitialization strategy to range for float tensors.

pub fn range_int(self, min: i64, max: i64) -> Self

Set the reinitialization strategy to range for int tensors.

Trait Implementations

impl<B: Debug + Backend> Debug for Reinitializer<B>

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

Formats the value using the given formatter.

impl<B: Backend> Default for Reinitializer<B>

fn default() -> Self

Returns the “default value” for a type.

impl<B: Backend> ModuleMapper<B> for Reinitializer<B>

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.