Struct Tensor 

pub struct Tensor<T>
where
    T: TensorElement,
{ /* private fields */ }

Implementations

impl<T> Tensor<T>

where T: TensorElement,

pub fn new(data: Box<[T]>, shape: Vec<usize>) -> Tensor<T>

pub fn shape(&self) -> &[usize]

pub fn size(&self) -> usize

pub fn flat_to_index(&self, index: usize) -> Vec<usize>

pub fn zeros(shape: &[usize]) -> Tensor<T>

pub fn ones(shape: &[usize]) -> Tensor<T>

pub fn eye(shape: &[usize]) -> Tensor<T>

pub fn diag(tensor: &Tensor<T>) -> Tensor<T>

pub fn scalar(value: T) -> Tensor<T>

pub fn is_scalar(&self) -> bool

pub fn extract_scalar(&self) -> Result<T, Error>

pub fn from_vec(data: Vec<T>, shape: Option<Vec<usize>>) -> Tensor<T>

pub fn to_vec(&self) -> Vec<T>

pub fn reshape(&mut self, new_shape: &[usize]) -> Result<(), Error>

pub fn transpose(&self) -> Tensor<T>

pub fn map(&self, f: impl Fn(T) -> T) -> Tensor<T>

pub fn apply_binary_op( &self, other: &Tensor<T>, f: impl Fn(T, T) -> T, ) -> Option<Tensor<T>>

pub fn reduce(&self, f: impl Fn(T, T) -> T) -> T

pub fn sub_tensor( &self, indices: Vec<Option<usize>>, ) -> Result<Tensor<T>, Error>

pub fn multi_index( &self, indices: Vec<Tensor<usize>>, ) -> Result<Tensor<T>, Error>

pub fn generalized_contraction( &self, other: &Tensor<T>, f: impl Fn(&Tensor<T>) -> T, g: impl Fn(T, T) -> T, ) -> Result<Tensor<T>, Error>

pub fn dot(&self, other: &Tensor<T>) -> Result<Tensor<T>, Error>

pub fn from_tensors(tensors: &[Tensor<T>]) -> Result<Tensor<T>, Error>

Concatenates a list of tensors of the same shape along a new axis

pub fn to_ndarray2(&self) -> ArrayBase<OwnedRepr<T>, Dim<[usize; 2]>>

pub fn from_ndarray<D>(array: ArrayBase<OwnedRepr<T>, D>) -> Tensor<T>

where D: Dimension,

pub fn to_ndarray(&self) -> ArrayBase<OwnedRepr<T>, Dim<IxDynImpl>>

pub fn range(start: T, end: T) -> Tensor<T>

impl<T> Tensor<T>

where T: TensorElement + Scalar + Lapack,

pub fn qr(&self) -> Result<(Tensor<T>, Tensor<T>), Error>

pub fn svd(&self) -> Result<(Tensor<T>, Tensor<T>, Tensor<T>), Error>

pub fn solve(&self, other: &Tensor<T>) -> Result<Tensor<T>, Error>

impl<T> Tensor<T>

where T: Clone + Float + PartialOrd,

pub fn randn(shape: &[usize]) -> Tensor<T>

pub fn randu(shape: &[usize]) -> Tensor<T>

Trait Implementations

impl<T> Clone for Tensor<T>

where T: Clone + TensorElement,

fn clone(&self) -> Tensor<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T> Display for Tensor<T>

where T: Display + TensorElement,

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

Formats the value using the given formatter.

impl<T> Index<&[usize]> for Tensor<T>

where T: TensorElement,

type Output = T

The returned type after indexing.

fn index(&self, index: &[usize]) -> &<Tensor<T> as Index<&[usize]>>::Output

Performs the indexing (container[index]) operation.

impl<T> IndexMut<&[usize]> for Tensor<T>

where T: TensorElement,

fn index_mut( &mut self, index: &[usize], ) -> &mut <Tensor<T> as Index<&[usize]>>::Output

Performs the mutable indexing (container[index]) operation.