Struct rai_core::Tensor

pub struct Tensor(/* private fields */);

Implementations

impl Tensor

pub fn new( backend: impl Into<Box<dyn Backend>>, dtype: DType, shape: impl Shape, primitive: impl Into<Box<dyn Primitive>>, inputs: impl Into<Vec<Tensor>> ) -> Self

pub fn id(&self) -> usize

pub fn backend(&self) -> &dyn Backend

pub fn dtype(&self) -> DType

pub fn shape(&self) -> impl Shape + '_

pub fn primitive(&self) -> &dyn Primitive

pub fn inputs(&self) -> impl Deref<Target = [Tensor]> + '_

pub fn full( val: f64, shape: impl Shape, dtype: DType, backend: impl Into<Box<dyn Backend>> + Debug ) -> Tensor

pub fn ones( shape: impl Shape, dtype: DType, backend: impl Into<Box<dyn Backend>> + Debug ) -> Tensor

pub fn zeros( shape: impl Shape, dtype: DType, backend: impl Into<Box<dyn Backend>> + Debug ) -> Tensor

pub fn full_like(&self, val: f64) -> Tensor

pub fn zeros_like(&self) -> Tensor

pub fn ones_like(&self) -> Tensor

pub fn normal( shape: impl Shape, dtype: DType, backend: impl Into<Box<dyn Backend>> + Debug ) -> Tensor

pub fn arange<T: ArangeArgs>( args: T, backend: impl Into<Box<dyn Backend>> + Debug ) -> Tensor

pub fn matmul<T: AsRef<Tensor>>(&self, rhs: T) -> Tensor

pub fn t(&self) -> Tensor

pub fn transpose(&self, axes: impl Into<Vec<usize>> + Debug) -> Tensor

pub fn broadcast_to(&self, shape: impl Shape) -> Tensor

pub fn reshape(&self, shape: impl Shape) -> Tensor

pub fn add<T>(&self, rhs: T) -> Tensor

where for<'a> &'a Self: Add<T, Output = Tensor>,

pub fn mul<T>(&self, rhs: T) -> Tensor

where for<'a> &'a Self: Mul<T, Output = Tensor>,

pub fn sin(&self) -> Tensor

pub fn cos(&self) -> Tensor

pub fn square(&self) -> Tensor

pub fn sqrt(&self) -> Tensor

pub fn rsqrt(&self) -> Tensor

pub fn sign(&self) -> Tensor

pub fn abs(&self) -> Tensor

pub fn exp(&self) -> Tensor

pub fn sum(&self) -> Tensor

pub fn reduce_sum<T: ReduceSumArgs>(&self, args: T) -> Tensor

pub fn jvp(&self, tangent_cache: &mut HashMap<usize, Tensor>) -> Tensor

pub fn vjp(&self, cotangent_cache: &mut HashMap<usize, Tensor>)

pub fn dot_graph(&self) -> String

pub fn detach(&self)

pub fn set_data<T: TensorLike + 'static>(&self, data: T)

pub fn get_data<T>(&self) -> Option<impl Deref<Target = T> + '_>

where T: 'static,

pub fn is_evalualted(&self) -> bool

Trait Implementations

impl Add<&Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: &Tensor) -> Tensor

Performs the + operation.

impl Add<&Tensor> for Tensor

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: &Tensor) -> Tensor

Performs the + operation.

impl Add<&Tensor> for f32

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: &Tensor) -> Self::Output

Performs the + operation.

impl Add<&Tensor> for f64

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: &Tensor) -> Self::Output

Performs the + operation.

impl<T> Add<T> for &Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: T) -> Self::Output

Performs the + operation.

impl<T> Add<T> for Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: T) -> Self::Output

Performs the + operation.

impl Add<Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: Tensor) -> Tensor

Performs the + operation.

impl Add<Tensor> for f32

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: Tensor) -> Self::Output

Performs the + operation.

impl Add<Tensor> for f64

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: Tensor) -> Self::Output

Performs the + operation.

impl Add for Tensor

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: Tensor) -> Tensor

Performs the + operation.

impl AsRef<Tensor> for Tensor

fn as_ref(&self) -> &Tensor

Converts this type into a shared reference of the (usually inferred) input type.

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Tensor

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

Formats the value using the given formatter.

impl Display for Tensor

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

Formats the value using the given formatter.

impl Div<&Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: &Tensor) -> Tensor

Performs the / operation.

impl Div<&Tensor> for Tensor

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: &Tensor) -> Tensor

Performs the / operation.

impl Div<&Tensor> for f32

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: &Tensor) -> Self::Output

Performs the / operation.

impl Div<&Tensor> for f64

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: &Tensor) -> Self::Output

Performs the / operation.

impl<T> Div<T> for &Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: T) -> Self::Output

Performs the / operation.

impl<T> Div<T> for Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: T) -> Self::Output

Performs the / operation.

impl Div<Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: Tensor) -> Tensor

Performs the / operation.

impl Div<Tensor> for f32

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: Tensor) -> Self::Output

Performs the / operation.

impl Div<Tensor> for f64

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: Tensor) -> Self::Output

Performs the / operation.

impl Div for Tensor

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: Tensor) -> Tensor

Performs the / operation.

impl Hash for Tensor

fn hash<H>(&self, hasher: &mut H)

where H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl Mul<&Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: &Tensor) -> Tensor

Performs the * operation.

impl Mul<&Tensor> for Tensor

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: &Tensor) -> Tensor

Performs the * operation.

impl Mul<&Tensor> for f32

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: &Tensor) -> Self::Output

Performs the * operation.

impl Mul<&Tensor> for f64

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: &Tensor) -> Self::Output

Performs the * operation.

impl<T> Mul<T> for &Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: T) -> Self::Output

Performs the * operation.

impl<T> Mul<T> for Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: T) -> Self::Output

Performs the * operation.

impl Mul<Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: Tensor) -> Tensor

Performs the * operation.

impl Mul<Tensor> for f32

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: Tensor) -> Self::Output

Performs the * operation.

impl Mul<Tensor> for f64

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: Tensor) -> Self::Output

Performs the * operation.

impl Mul for Tensor

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: Tensor) -> Tensor

Performs the * operation.

impl Neg for &Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn neg(self) -> Self::Output

Performs the unary - operation.

impl Neg for Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn neg(self) -> Self::Output

Performs the unary - operation.

impl Ord for Tensor

fn cmp(&self, other: &Self) -> Ordering

This method returns an Ordering between self and other.

fn max(self, other: Self) -> Self

where Self: Sized,

Compares and returns the maximum of two values.

fn min(self, other: Self) -> Self

where Self: Sized,

Compares and returns the minimum of two values.

fn clamp(self, min: Self, max: Self) -> Self

where Self: Sized + PartialOrd,

Restrict a value to a certain interval.

impl PartialEq for Tensor

fn eq(&self, other: &Self) -> 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 PartialOrd for Tensor

fn partial_cmp(&self, other: &Self) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

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

This method tests less than (for self and other) and is used by the < operator.

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

This method tests less than or equal to (for self and other) and is used by the <= operator.

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

This method tests greater than (for self and other) and is used by the > operator.

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

This method tests greater than or equal to (for self and other) and is used by the >= operator.

impl Shape for &Tensor

fn dims(&self) -> &[usize]

fn ndim(&self) -> usize

fn size(&self) -> usize

fn to_vec(&self) -> Vec<usize>

fn shape_transpose(&self) -> Vec<usize>

fn shape_at(&self, index: usize) -> usize

where Self: Sized,

fn shape_eq(&self, rhs: &impl Shape) -> bool

where Self: Sized,

fn shape_ndim_eq(&self, rhs: &impl Shape) -> bool

where Self: Sized,

fn shape_size_eq(&self, rhs: &impl Shape) -> bool

where Self: Sized,

fn shape_broadcast(&self, rhs: &impl Shape) -> Result<Vec<usize>>

where Self: Sized,

fn shape_broadcast_matmul(&self, rhs: &impl Shape) -> Result<Vec<usize>>

where Self: Sized,

impl Shape for Tensor

fn dims(&self) -> &[usize]

fn ndim(&self) -> usize

fn size(&self) -> usize

fn to_vec(&self) -> Vec<usize>

fn shape_transpose(&self) -> Vec<usize>

fn shape_at(&self, index: usize) -> usize

where Self: Sized,

fn shape_eq(&self, rhs: &impl Shape) -> bool

where Self: Sized,

fn shape_ndim_eq(&self, rhs: &impl Shape) -> bool

where Self: Sized,

fn shape_size_eq(&self, rhs: &impl Shape) -> bool

where Self: Sized,

fn shape_broadcast(&self, rhs: &impl Shape) -> Result<Vec<usize>>

where Self: Sized,

fn shape_broadcast_matmul(&self, rhs: &impl Shape) -> Result<Vec<usize>>

where Self: Sized,

impl Sub<&Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: &Tensor) -> Tensor

Performs the - operation.

impl Sub<&Tensor> for Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: &Tensor) -> Tensor

Performs the - operation.

impl Sub<&Tensor> for f32

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: &Tensor) -> Self::Output

Performs the - operation.

impl Sub<&Tensor> for f64

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: &Tensor) -> Self::Output

Performs the - operation.

impl<T> Sub<T> for &Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: T) -> Self::Output

Performs the - operation.

impl<T> Sub<T> for Tensor

where T: 'static + ElemType,

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: T) -> Self::Output

Performs the - operation.

impl Sub<Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: Tensor) -> Tensor

Performs the - operation.

impl Sub<Tensor> for f32

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: Tensor) -> Self::Output

Performs the - operation.

impl Sub<Tensor> for f64

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: Tensor) -> Self::Output

Performs the - operation.

impl Sub for Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: Tensor) -> Tensor

Performs the - operation.

impl TensorIter for &Tensor

fn tensor_iter(&self) -> impl Iterator<Item = &Tensor>

impl TensorIter for Tensor

fn tensor_iter(&self) -> impl Iterator<Item = &Tensor>

impl WithTensors for &Tensor

fn tensors(&self) -> Vec<Tensor>

impl WithTensors for Tensor

fn tensors(&self) -> Vec<Tensor>

impl Eq for Tensor