Struct rai_core::primitives::Arange

pub struct Arange<D: Type> {
    pub start: D::Repr,
    pub stop: D::Repr,
    pub step: D::Repr,
}

Fields

start: D::Repr

stop: D::Repr

step: D::Repr

Implementations

impl<D: Type> Arange<D>

pub fn new(start: D::Repr, stop: D::Repr, step: D::Repr) -> Self

Trait Implementations

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

where D::Repr: Clone,

fn clone(&self) -> Arange<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 Arange<D>

where D::Repr: Debug,

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

Formats the value using the given formatter.

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

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

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

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

where D::Repr: PartialEq,

fn eq(&self, other: &Arange<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 Arange<D>

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<D: Type> StructuralPartialEq for Arange<D>