Struct rai_core::Tensor

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

Implementations

impl Tensor

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

pub fn from_safetensor( st: &TensorView<'_>, backend: impl Into<Box<dyn Backend>> ) -> Tensor

pub fn id(&self) -> usize

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

pub fn dtype(&self) -> &dyn DynDType

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

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

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

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

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

pub fn full_like<T: ElemType>(&self, val: T) -> Tensor

pub fn zeros_like(&self) -> Tensor

pub fn ones_like(&self) -> Tensor

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

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

where D::Repr: Sub<D::Repr, Output = D::Repr> + Div<D::Repr, Output = D::Repr> + Into<f64> + Copy,

pub fn from_array<T: ElemType>( data: impl Into<Vec<T>> + Debug, shape: impl Shape, backend: impl Into<Box<dyn Backend>> + Debug ) -> Tensor

pub fn neg(&self) -> Tensor

pub fn cat<T: AsRef<Tensor> + Debug>(tensors: &[T], dim: impl Dim) -> 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, dim0: impl Dim, dim1: impl Dim) -> Tensor

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

pub fn broadcast_left(&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 tanh(&self) -> Tensor

pub fn square(&self) -> Tensor

pub fn powf(&self, exponent: f64) -> 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 var<T: VarArgs>(&self, args: T) -> Tensor

pub fn argmax<T: ArgReduceArgs>(&self, args: T) -> Tensor

pub fn argmin<T: ArgReduceArgs>(&self, args: T) -> Tensor

pub fn gather(&self, dim: impl Dim, index: &Tensor) -> Tensor

pub fn index_select(&self, dim: impl Dim, index: &Tensor) -> Tensor

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

pub fn as_type_of(&self, rhs: &Tensor) -> Tensor

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

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

pub fn erf(&self) -> Tensor

pub fn relu(&self) -> Tensor

pub fn gelu(&self) -> Tensor

pub fn new_gelu(&self) -> Tensor

pub fn flatten<T: FlattenArgs>(&self, args: T) -> Tensor

pub fn to_contiguous(&self) -> Tensor

pub fn squeeze(&self, d: impl Dims) -> Tensor

pub fn unsqueeze(&self, d: impl Dim) -> Tensor

pub fn narrow(&self, dim: impl Dim, start: usize, len: usize) -> Tensor

pub fn chunk(&self, chunks: usize, dim: impl Dim) -> Vec<Tensor>

pub fn where_cond(&self, input: &Tensor, other: &Tensor) -> 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 as_scalar<T: ElemType>(&self) -> T

pub fn as_vec<T: ElemType>(&self) -> Vec<T>

pub fn is_evaluated(&self) -> bool

pub fn to_safetensors(&self, name: impl Into<String>, filename: &Path)

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> Add<&'a Tensor> for u8

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: 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: 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<Tensor> for u8

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> Div<&'a Tensor> for u8

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: 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: 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<Tensor> for u8

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 GenericValue<BasicValue, Tensor, Tensor> for Tensor

fn gv_tensors(&self) -> Tensor

fn gv_grad(tensor: &Tensor, grad_map: &HashMap<usize, Tensor>) -> Tensor

fn gv_grad_map(tensor: &Tensor, grad: Tensor, out: &mut HashMap<usize, 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> Mul<&'a Tensor> for u8

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: 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: 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<Tensor> for u8

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<'a> Neg for &'a 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 shape_of<D: Dims>(&self, d: D) -> Vec<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, dim0: usize, dim1: usize) -> 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_expand_left<T: Shape + ?Sized>(&self, rhs: &T) -> Vec<usize>

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> Sub<&'a Tensor> for u8

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: 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: 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<Tensor> for u8

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 ValueSpec for Tensor

type Kind = BasicValue

type Tensors = Tensor

type Gradient = Tensor

impl Eq for Tensor