pub enum ReduceOp {
Sum,
Max,
Mean,
}Expand description
Reduction operations: reduce tensor to scalar
These operations collapse all dimensions and require special gradient handling since the output shape differs from the input.
Variants§
Trait Implementations§
Auto Trait Implementations§
impl Freeze for ReduceOp
impl RefUnwindSafe for ReduceOp
impl Send for ReduceOp
impl Sync for ReduceOp
impl Unpin for ReduceOp
impl UnwindSafe for ReduceOp
Blanket Implementations§
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