Skip to main content

EagerTensorEinsumExt

Trait EagerTensorEinsumExt 

Source
pub trait EagerTensorEinsumExt {
    // Required method
    fn tensordot(
        &self,
        rhs: &EagerTensor,
        axes: TensorDotAxes<'_>,
    ) -> Result<EagerTensor>;
}
Expand description

Eager tensor contraction-sugar methods.

Required Methods§

Source

fn tensordot( &self, rhs: &EagerTensor, axes: TensorDotAxes<'_>, ) -> Result<EagerTensor>

Contract two eager tensors over the requested axes.

§Errors

Returns Error::Validation with AxisOutOfBounds, DuplicateAxis, RankMismatch, or ShapeMismatch for invalid axes/shapes, or Error::Runtime for backend execution failures.

Implementations on Foreign Types§

Source§

impl EagerTensorEinsumExt for EagerTensor

Source§

fn tensordot( &self, rhs: &EagerTensor, axes: TensorDotAxes<'_>, ) -> Result<EagerTensor>

Implementors§