Trait GraphCompilerEinsumExt 

pub trait GraphCompilerEinsumExt {
    // Required methods
    fn einsum(
        &mut self,
        inputs: &[&TracedTensor],
        subscripts: &str,
    ) -> Result<TracedTensor>;
    fn einsum_subscripts(
        &mut self,
        inputs: &[&TracedTensor],
        subscripts: &EinsumSubscripts,
    ) -> Result<TracedTensor>;
    fn einsum_with(
        &mut self,
        inputs: &[&TracedTensor],
        subscripts: &str,
        optimize: EinsumOptimize,
    ) -> Result<TracedTensor>;
    fn einsum_subscripts_with(
        &mut self,
        inputs: &[&TracedTensor],
        subscripts: &EinsumSubscripts,
        optimize: EinsumOptimize,
    ) -> Result<TracedTensor>;
}

Traced einsum extension methods for GraphCompiler.

Required Methods

fn einsum( &mut self, inputs: &[&TracedTensor], subscripts: &str, ) -> Result<TracedTensor>

fn einsum_subscripts( &mut self, inputs: &[&TracedTensor], subscripts: &EinsumSubscripts, ) -> Result<TracedTensor>

fn einsum_with( &mut self, inputs: &[&TracedTensor], subscripts: &str, optimize: EinsumOptimize, ) -> Result<TracedTensor>

fn einsum_subscripts_with( &mut self, inputs: &[&TracedTensor], subscripts: &EinsumSubscripts, optimize: EinsumOptimize, ) -> Result<TracedTensor>

Implementations on Foreign Types

impl GraphCompilerEinsumExt for GraphCompiler

fn einsum( &mut self, inputs: &[&TracedTensor], subscripts: &str, ) -> Result<TracedTensor>

fn einsum_subscripts( &mut self, inputs: &[&TracedTensor], subscripts: &EinsumSubscripts, ) -> Result<TracedTensor>

fn einsum_with( &mut self, inputs: &[&TracedTensor], subscripts: &str, optimize: EinsumOptimize, ) -> Result<TracedTensor>

fn einsum_subscripts_with( &mut self, inputs: &[&TracedTensor], subscripts: &EinsumSubscripts, optimize: EinsumOptimize, ) -> Result<TracedTensor>

Implementors