Trait rai_core::primitives::Primitive
source · pub trait Primitive: Debug {
// Required methods
fn clone_boxed(&self) -> Box<dyn Primitive>;
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 { ... }
}