Struct rai_core::primitives::Normal

pub struct Normal<D: Type> {
    pub mean: D::Repr,
    pub std: D::Repr,
}

Fields

mean: D::Repr

std: D::Repr

Implementations

impl<D: Type> Normal<D>

pub fn new(mean: D::Repr, std: D::Repr) -> Self

Trait Implementations

impl<D: Clone + Type> Clone for Normal<D>

where D::Repr: Clone,

fn clone(&self) -> Normal<D>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<D: Debug + Type> Debug for Normal<D>

where D::Repr: Debug,

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<D, T> Eval<D, Normal<T>> for CandleBackend

where D: Device, for<'a> &'a D: Into<Device>, T: Type, T::Repr: FloatDType,

fn eval(&self, device: &D, primitive: &Normal<T>, _: &[Tensor], output: &Tensor)

impl<D: PartialEq + Type> PartialEq for Normal<D>

where D::Repr: PartialEq,

fn eq(&self, other: &Normal<D>) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<D: Type> Primitive for Normal<D>

fn clone_boxed(&self) -> Box<dyn Primitive>

fn as_any(&self) -> &dyn Any

fn dot_label(&self) -> String

fn jvp( &self, output: &Tensor, _primals: &[Tensor], _tangents: &[Tensor] ) -> Tensor

fn vjp( &self, _output: &Tensor, _primals: &[Tensor], _cotangent: &Tensor ) -> Vec<Tensor>

impl<D: Copy + Type> Copy for Normal<D>

where D::Repr: Copy,

impl<D: Eq + Type> Eq for Normal<D>

where D::Repr: Eq,

impl<D: Type> StructuralPartialEq for Normal<D>