Struct EinsumPlan 

pub struct EinsumPlan { /* private fields */ }

A reusable einsum expression with fixed input shapes, strides and storage types.

Implementations

impl EinsumPlan

pub fn new(expression: &str, inputs: &[TensorDescriptor]) -> Result<Self>

pub fn with_options( expression: &str, inputs: &[TensorDescriptor], output_dtype: DType, compute: ComputeType, ) -> Result<Self>

pub fn expression(&self) -> &str

pub fn plan(&self) -> &Plan

pub fn output(&self) -> &TensorDescriptor

pub fn optimize(self) -> Result<Self>

Choose a minimum-product binary contraction tree by exhaustive subset dynamic programming. Search is exponential in operand count. Reassociation changes floating-point operation order; intermediates use the declared compute dtype and only the final output is cast to its dtype.

pub fn planned_scalar_products(&self) -> Option<u128>

pub fn execute<R: Runtime>( &self, inputs: &[&RudaTensor<R>], ) -> Result<RudaTensor<R>>

pub fn execute_into<R: Runtime>( &self, inputs: &[&RudaTensor<R>], output: RudaTensor<R>, ) -> Result<RudaTensor<R>>

Trait Implementations

impl Clone for EinsumPlan

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for EinsumPlan

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.