Struct rai_core::primitives::ArgMin
source · pub struct ArgMin {
pub dim: usize,
pub keep_dim: bool,
}
Fields§
§dim: usize
§keep_dim: bool
Implementations§
Trait Implementations§
source§impl PartialEq for ArgMin
impl PartialEq for ArgMin
source§impl Primitive for ArgMin
impl Primitive for ArgMin
fn clone_boxed(&self) -> Box<dyn Primitive>
fn dot_label(&self) -> String
fn as_any(&self) -> &dyn Any
fn jvp( &self, output: &Tensor, primals: &[Tensor], tangents: &[Tensor] ) -> Tensor
fn vjp( &self, output: &Tensor, primals: &[Tensor], cotangent: &Tensor ) -> Vec<Tensor>
impl Eq for ArgMin
impl StructuralPartialEq for ArgMin
Auto Trait Implementations§
impl Freeze for ArgMin
impl RefUnwindSafe for ArgMin
impl Send for ArgMin
impl Sync for ArgMin
impl Unpin for ArgMin
impl UnwindSafe for ArgMin
Blanket Implementations§
source§impl<T, M> ApplyModule<M> for Twhere
M: Module<Input = T>,
impl<T, M> ApplyModule<M> for Twhere
M: Module<Input = 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