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 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 eq<T: AsRef<Tensor>>(&self, rhs: T) -> Tensor

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

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

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

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

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

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

pub fn t(&self) -> Tensor

pub fn transpose(&self, dims: 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 log(&self) -> Tensor

pub fn log2(&self) -> Tensor

pub fn log10(&self) -> Tensor

pub fn sum<T: ReduceArgs>(&self, args: T) -> Tensor

pub fn max<T: ReduceArgs>(&self, args: T) -> Tensor

pub fn min<T: ReduceArgs>(&self, args: T) -> Tensor

pub fn mean<T: ReduceArgs>(&self, args: T) -> Tensor

pub fn as_type(&self, dtype: DType) -> Tensor

pub fn softmax<D: Dim>(&self, d: D) -> Tensor

pub fn log_softmax<D: Dim>(&self, d: D) -> Tensor

pub fn relu(&self) -> 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 replace_data(&self, rhs: Tensor)

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_evaluated(&self) -> bool

Trait Implementations

impl<'a, 'b> Add<&'b Tensor> for &'a Tensor

type Output = Tensor

The resulting type after applying the + operator.

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

Performs the + operation.

impl<'a> Add<&'a Tensor> for Tensor

type Output = Tensor

The resulting type after applying the + operator.

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

Performs the + operation.

impl<'a> Add<&'a Tensor> for f32

type Output = Tensor

The resulting type after applying the + operator.

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

Performs the + operation.

impl<'a> Add<&'a Tensor> for f64

type Output = Tensor

The resulting type after applying the + operator.

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

Performs the + operation.

impl<'a, T> Add<T> for &'a 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<'a> Add<Tensor> for &'a 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<'a, 'b> Div<&'b Tensor> for &'a Tensor

type Output = Tensor

The resulting type after applying the / operator.

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

Performs the / operation.

impl<'a> Div<&'a Tensor> for Tensor

type Output = Tensor

The resulting type after applying the / operator.

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

Performs the / operation.

impl<'a> Div<&'a Tensor> for f32

type Output = Tensor

The resulting type after applying the / operator.

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

Performs the / operation.

impl<'a> Div<&'a Tensor> for f64

type Output = Tensor

The resulting type after applying the / operator.

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

Performs the / operation.

impl<'a, T> Div<T> for &'a 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<'a> Div<Tensor> for &'a 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<'m, 'i, M, F> Func<(&'m M, &'i Tensor), Tensor> for F

where M: Module, F: Fn(&'m M, &'i Tensor) -> Tensor,

type Tangent = HashMap<usize, Tensor>

type Cotangent = HashMap<usize, Tensor>

fn call(&self, input: (&'m M, &'i Tensor)) -> Tensor

fn extract_input_tensors( &self, input: &(&'m M, &'i Tensor), inputs: &mut Vec<Tensor> )

fn self_captured_tensors(&self, _tensors: &mut Vec<Tensor>)

impl<'m, 'i, 'l, M, F> Func<(&'m M, &'i Tensor, &'l Tensor), Tensor> for F

where M: Module, F: Fn(&'m M, &'i Tensor, &'l Tensor) -> Tensor,

type Tangent = HashMap<usize, Tensor>

type Cotangent = HashMap<usize, Tensor>

fn call(&self, input: (&'m M, &'i Tensor, &'l Tensor)) -> Tensor

fn extract_input_tensors( &self, input: &(&'m M, &'i Tensor, &'l Tensor), inputs: &mut Vec<Tensor> )

fn self_captured_tensors(&self, _tensors: &mut Vec<Tensor>)

impl<T> Func<Tensor, Tensor> for T

where T: Module,

type Tangent = HashMap<usize, Tensor>

type Cotangent = HashMap<usize, Tensor>

fn call(&self, input: Tensor) -> Tensor

fn self_captured_tensors(&self, tensors: &mut Vec<Tensor>)

fn extract_input_tensors(&self, _input: &Tensor, _tensors: &mut Vec<Tensor>)

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<'a, 'b> Mul<&'b Tensor> for &'a Tensor

type Output = Tensor

The resulting type after applying the * operator.

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

Performs the * operation.

impl<'a> Mul<&'a Tensor> for Tensor

type Output = Tensor

The resulting type after applying the * operator.

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

Performs the * operation.

impl<'a> Mul<&'a Tensor> for f32

type Output = Tensor

The resulting type after applying the * operator.

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

Performs the * operation.

impl<'a> Mul<&'a Tensor> for f64

type Output = Tensor

The resulting type after applying the * operator.

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

Performs the * operation.

impl<'a, T> Mul<T> for &'a 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<'a> Mul<Tensor> for &'a 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 shape(&self) -> &[usize]

fn ndim(&self) -> usize

fn size(&self) -> usize

fn shape_at<I: Dim>(&self, i: I) -> usize

fn shape_until<I: Dim>(&self, i: I) -> &[usize]

fn dim<I: Dim>(&self, i: I) -> usize

fn dims<D: Dims>(&self, d: D) -> Vec<usize>

fn size_of_dims<D: Dims>(&self, d: D) -> usize

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

fn shape_eq<S: Shape + ?Sized>(&self, rhs: &S) -> bool

fn shape_ndim_eq<S: Shape + ?Sized>(&self, rhs: &S) -> bool

fn shape_size_eq<S: Shape + ?Sized>(&self, rhs: &S) -> bool

fn shape_broadcast<T: Shape + ?Sized>(&self, rhs: &T) -> Result<Vec<usize>>

fn shape_broadcast_matmul<S: Shape + ?Sized>( &self, rhs: &S ) -> Result<Vec<usize>>

fn shape_reduce<T: AsRef<[usize]>>(&self, dims: T, keep_dim: bool) -> Vec<usize>

impl<'a, 'b> Sub<&'b Tensor> for &'a Tensor

type Output = Tensor

The resulting type after applying the - operator.

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

Performs the - operation.

impl<'a> Sub<&'a Tensor> for Tensor

type Output = Tensor

The resulting type after applying the - operator.

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

Performs the - operation.

impl<'a> Sub<&'a Tensor> for f32

type Output = Tensor

The resulting type after applying the - operator.

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

Performs the - operation.

impl<'a> Sub<&'a Tensor> for f64

type Output = Tensor

The resulting type after applying the - operator.

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

Performs the - operation.

impl<'a, T> Sub<T> for &'a 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<'a> Sub<Tensor> for &'a 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