Struct Transaction

pub struct Transaction<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: Backend> Transaction<B>

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

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 + Backend> Default for Transaction<B>

fn default() -> Transaction<B>

Returns the “default value” for a type.