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, _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.