Struct Transaction

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

A transaction can read multiple tensors at once with a single operation improving compute utilization with optimized laziness.

Example

 let [output_data, loss_data, targets_data] = Transaction::default()
   .register(output)
   .register(loss)
   .register(targets)
   .execute()
   .try_into()
   .expect("Correct amount of tensor data");

Implementations

impl<B> Transaction<B>

where B: Backend,

pub fn register<const D: usize, K>( self, tensor: Tensor<B, D, K>, ) -> Transaction<B>

where K: BasicOps<B>,

Add a tensor to the transaction to be read.

pub fn execute(self) -> Vec<TensorData>

Executes the transaction synchronously and returns the data in the same order in which they were registered.

pub fn execute_async(self) -> impl Future<Output = Vec<TensorData>>

Executes the transaction asynchronously and returns the data in the same order in which they were registered.

Trait Implementations

impl<B> Default for Transaction<B>

where B: Default + Backend,

fn default() -> Transaction<B>

Returns the “default value” for a type.