Module tensor

Implementations for tensor operations and structures

Structs

Tensor

A multidimensional data structure not unlike ndarray::ArrayBase.

Functions

axpy_ctx

Performs the basic vector operation.

gemm

Performs the basic matmul operation.

gemm_ctx

Performs the basic matmul operation.

gemm_uninit

Performs the basic matmul operation.

gemm_uninit_ctx

Performs the basic matmul operation.

gemv

Performs the basic matrix-vector multiplication operation.

gemv_ctx

Performs the basic matrix-vector multiplication operation.

gemv_uninit

Performs the basic matrix-vector multiplication operation.

gemv_uninit_ctx

Performs the basic matrix-vector multiplication operation.

scal_ctx

Performs the basic vector operation.

Type Aliases

Matrix

A 2-dimensional tensor

MatrixView

A ‘view’ of a matrix, Like &[T] is to Vec<T>

MatrixViewMut

A ‘mut view’ of a matrix, Like &mut [T] is to Vec<T>

Slice

A representation of a slice

TensorView

A ‘view’ of a tensor, Like &[T] is to Vec<T>

TensorViewMut

A ‘mut view’ of a tensor, Like &mut [T] is to Vec<T>

UninitMatrix

An uninit matrix. Contents are mutable and specified as MaybeUninit.

UninitTensor

An uninit tensor. Contents are mutable and specified as MaybeUninit.

UninitVector

An uninit vector. Contents are mutable and specified as MaybeUninit.

Vector

A 1-dimensional tensor

VectorView

A ‘view’ of a vector, Like &[T] is to Vec<T>

VectorViewMut

A ‘mut view’ of a vector, Like &mut [T] is to Vec<T>

ViewOf

Gets the view repr of the provided storage