Struct EpsilonInjectorWrapper 

pub struct EpsilonInjectorWrapper<Inner: InfererWrapper, NG: NoiseGenerator> { /* private fields */ }

Implementations

impl<Inner: InfererWrapper> EpsilonInjectorWrapper<Inner, HighQualityNoiseGenerator>

pub fn wrap( inner: Inner, inferer: &dyn Inferer, key: &str, ) -> Result<EpsilonInjectorWrapper<Inner, HighQualityNoiseGenerator>>

Wraps the provided inferer to automatically generate noise for the input named by key.

This function will use HighQualityNoiseGenerator as the noise source.

Errors

Will return an error if the provided key doesn’t match an input on the model.

impl<Inner, NG> EpsilonInjectorWrapper<Inner, NG>

where Inner: InfererWrapper, NG: NoiseGenerator,

pub fn with_generator( inner: Inner, inferer: &dyn Inferer, generator: NG, key: &str, ) -> Result<Self>

Create a new injector for the provided key, using the custom generator as the noise source.

Errors

Will return an error if the provided key doesn’t match an input on the model.

Trait Implementations

impl<Inner, NG> InfererWrapper for EpsilonInjectorWrapper<Inner, NG>

where Inner: InfererWrapper, NG: NoiseGenerator,

fn invoke( &self, inferer: &dyn Inferer, batch: &mut ScratchPadView<'_>, ) -> Result<()>

Invokes the inner inferer, applying any additional logic before and after the call.

fn input_shapes<'a>( &'a self, _inferer: &'a dyn Inferer, ) -> &'a [(String, Vec<usize>)]

Returns the input shapes after this wrapper has been applied.

fn output_shapes<'a>( &'a self, inferer: &'a dyn Inferer, ) -> &'a [(String, Vec<usize>)]

Returns the output shapes after this wrapper has been applied.

fn begin_agent(&self, inferer: &dyn Inferer, id: u64)

Called when starting inference for a new agent.

fn end_agent(&self, inferer: &dyn Inferer, id: u64)

Called when finishing inference for an agent.