Enum TensorDotDims

pub enum TensorDotDims<'a> {
    Int(i32),
    List((&'a [i32], &'a [i32])),
}

The number of dimensions to sum over

Variants

Int(i32)

Sum over the last axes dimensions of a and the first axes dimensions of b.

List((&'a [i32], &'a [i32]))

Sum over the corresponding dimensions of a and b.

Trait Implementations

impl<'a> Debug for TensorDotDims<'a>

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

Formats the value using the given formatter.

impl<'a> From<(&'a [i32], &'a [i32])> for TensorDotDims<'a>

fn from((lhs, rhs): (&'a [i32], &'a [i32])) -> Self

Converts to this type from the input type.

impl<'a, const M: usize, const N: usize> From<(&'a [i32; M], &'a [i32; N])> for TensorDotDims<'a>

fn from((lhs, rhs): (&'a [i32; M], &'a [i32; N])) -> Self

Converts to this type from the input type.

impl From<i32> for TensorDotDims<'_>

fn from(i: i32) -> Self

Converts to this type from the input type.