Skip to main content

TensorTensordotExt

Trait TensorTensordotExt 

Source
pub trait TensorTensordotExt {
    // Required method
    fn tensordot(
        &self,
        rhs: &Tensor,
        axes: TensorDotAxes<'_>,
        session: &mut dyn BackendSession,
    ) -> Result<Tensor>;
}
Expand description

Backend-explicit tensordot sugar for dtype-erased concrete tensors.

Required Methods§

Source

fn tensordot( &self, rhs: &Tensor, axes: TensorDotAxes<'_>, session: &mut dyn BackendSession, ) -> Result<Tensor>

Contract this tensor with rhs over explicit axes or an axis count.

§Errors

Returns Error::Validation with InvalidArgument, RankMismatch, AxisOutOfBounds, or ShapeMismatch for invalid axes or incompatible contracting dimensions, or Error::Tensor when backend execution fails.

Dyn Compatibility§

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§