Struct LayerwisePolicyForward 

pub struct LayerwisePolicyForward<'a, B, U, P>
where
    B: NeuralBackend,
    P: LayerwisePolicy<B, U>,
{ /* private fields */ }

Failure-safe ownership of one policy forward and its current unit lease.

Implementations

impl<'a, B, U, P> LayerwisePolicyForward<'a, B, U, P>

where B: NeuralBackend, P: LayerwisePolicy<B, U>,

pub fn begin( policy: &'a mut P, initial: &B::Tensor, context: &'a <B::Tensor as Tensor>::Context, ) -> Result<Self, P::Error>

Begins one failure-safe policy transaction.

pub fn acquire<E, F>( &mut self, ordinal: usize, address: ExecutionUnitAddress, build: F, ) -> Result<&mut P::Lease, LayerwiseAcquireError<E, P::Error>>

where F: FnOnce(&<B::Tensor as Tensor>::Context) -> Result<U, E>,

Acquires one unit and retains its lease until completion or abort.

pub fn complete<'value, StateValues, ContextValues>( &mut self, output: &'value B::Tensor, state_values: StateValues, context_values: ContextValues, ) -> Result<(), P::Error>

where B::Tensor: 'value, StateValues: Iterator<Item = &'value B::Tensor>, ContextValues: Iterator<Item = &'value B::Tensor>,

Completes and returns the currently active unit lease.

pub fn finish(&mut self, output: &B::Tensor) -> Result<(), P::Error>

Completes the transaction; dropping before this call aborts it.

Trait Implementations

impl<B, U, P> Drop for LayerwisePolicyForward<'_, B, U, P>

where B: NeuralBackend, P: LayerwisePolicy<B, U>,

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.