Trait rai_core::primitives::Primitive

pub trait Primitive: Debug + DynClone {
    // Required methods
    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>;

    // Provided method
    fn dot_label(&self) -> String { ... }
}

Required Methods

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>

Provided Methods

fn dot_label(&self) -> String

Trait Implementations

impl<B, P> Eval<dyn Backend, dyn Primitive> for Dispatch<B, P>

where B: Backend + 'static + Sync + Send, P: Primitive + 'static + Sync + Send, Self: Eval<B, P> + 'static,

fn eval( &self, backend: &dyn Backend, primitive: &dyn Primitive, inputs: &[Tensor], output: &Tensor )

impl From<&Box<dyn Primitive>> for Box<dyn Primitive>

fn from(value: &Box<dyn Primitive>) -> Self

Converts to this type from the input type.

impl<T> From<T> for Box<dyn Primitive>

where T: Primitive + 'static,

fn from(t: T) -> Self

Converts to this type from the input type.

Implementors

impl Primitive for Add

impl Primitive for Broadcast

impl Primitive for Cos

impl Primitive for Div

impl Primitive for Full

impl Primitive for MatMul

impl Primitive for Mul

impl Primitive for Negative

impl Primitive for Normal

impl Primitive for ReduceSum

impl Primitive for Reshape

impl Primitive for Sin

impl Primitive for Square

impl Primitive for Sub

impl Primitive for Transpose