pub enum ShapeOp {
Add,
MatMul,
Reshape,
Transpose,
Concat,
Slice,
Broadcast,
}Expand description
Supported tensor operations for shape inference.
Variants§
Add
Element-wise addition (or any binary broadcast-compatible op).
MatMul
Matrix multiplication: (M,K) x (K,N) -> (M,N).
Reshape
Reshape to new dimensions (total elements must match).
Transpose
Reverse all dimensions.
Concat
Concatenate along an axis.
Slice
Slice along an axis.
Broadcast
Broadcast (expand) to a target shape.
Trait Implementations§
impl Copy for ShapeOp
impl Eq for ShapeOp
impl StructuralPartialEq for ShapeOp
Auto Trait Implementations§
impl Freeze for ShapeOp
impl RefUnwindSafe for ShapeOp
impl Send for ShapeOp
impl Sync for ShapeOp
impl Unpin for ShapeOp
impl UnsafeUnpin for ShapeOp
impl UnwindSafe for ShapeOp
Blanket Implementations§
impl<T> Allocation for T
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Compare self to
key and return true if they are equal.Source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more