Struct Tensor 

pub struct Tensor<Shape: TensorShape>(/* private fields */);

Implementations

impl<Shape> Tensor<Shape>

where Shape: TensorShape,

pub fn from_flat(data: [Float; Shape::SIZE]) -> Self

pub fn from_elem(value: Float) -> Self

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn rank(&self) -> usize

pub fn as_slice(&self) -> &[Float]

pub fn as_mut_slice(&mut self) -> &mut [Float]

pub fn at(&self, index: [usize; Shape::RANK]) -> &Float

pub fn set(&mut self, index: [usize; Shape::RANK], value: Float)

pub fn fill(&mut self, value: Float)

pub fn zeros() -> Self

pub fn random() -> Self

pub fn random_with_seed(seed: u64) -> Self

pub fn random_with<R>(rng: &mut R) -> Self

where R: Rng + ?Sized,

pub fn reshape<NewShape>(self) -> Tensor<NewShape>

where NewShape: TensorShape, (): ReshapePreservesElementCount<{ Shape::SIZE }, { NewShape::SIZE }>,

pub fn as_ref(&self) -> TensorRef<'_, Shape>

pub fn as_mut(&mut self) -> TensorMut<'_, Shape>

pub fn map_inplace<F>(&mut self, f: F)

where F: FnMut(Float) -> Float,

pub fn map<F>(&self, f: F) -> Self

where F: FnMut(Float) -> Float,

pub fn zip_map<F>(&self, rhs: &Self, f: F) -> Self

where F: FnMut(Float, Float) -> Float,

pub fn sum(&self) -> Float

pub fn mean(&self) -> Float

pub fn slice<T: Iterator>(_range: T)

👎Deprecated:

Tensor::slice is not implemented yet

impl<Shape> Tensor<Shape>

where Shape: NonScalarShape,

pub fn get_ref(&self, index: usize) -> TensorRef<'_, Shape::Subshape>

pub fn get_mut(&mut self, index: usize) -> TensorMut<'_, Shape::Subshape>

pub fn get(&self, index: usize) -> Tensor<Shape::Subshape>

impl<const N: usize> Tensor<Dim<N, Nil>>

pub fn dot(&self, rhs: &Self) -> Float

impl<const ROWS: usize, const COLS: usize> Tensor<Dim<ROWS, Dim<COLS, Nil>>>

pub fn transpose(&self) -> Tensor<Dim<COLS, Dim<ROWS, Nil>>>

pub fn matvec(&self, rhs: &Tensor<Dim<COLS, Nil>>) -> Tensor<Dim<ROWS, Nil>>

pub fn matmul<const OUT_COLS: usize>( &self, rhs: &Tensor<Dim<COLS, Dim<OUT_COLS, Nil>>>, ) -> Tensor<Dim<ROWS, Dim<OUT_COLS, Nil>>>

Trait Implementations

impl<Shape> Add<&Tensor<Shape>> for &Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the + operator.

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

Performs the + operation.

impl<Shape> Add<&Tensor<Shape>> for Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the + operator.

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

Performs the + operation.

impl<Shape> Add<f64> for Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the + operator.

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

Performs the + operation.

impl<Shape> AddAssign<&Tensor<Shape>> for Tensor<Shape>

where Shape: TensorShape,

fn add_assign(&mut self, rhs: &Tensor<Shape>)

Performs the += operation.

impl<Shape> AddAssign<f64> for Tensor<Shape>

where Shape: TensorShape,

fn add_assign(&mut self, rhs: Float)

Performs the += operation.

impl<Shape> Clone for Tensor<Shape>

where Shape: TensorShape,

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<Shape> Debug for Tensor<Shape>

where Shape: TensorShape,

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

Formats the value using the given formatter.

impl<Shape> Default for Tensor<Shape>

where Shape: TensorShape,

fn default() -> Self

Returns the “default value” for a type.

impl<Shape> Div<f64> for Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the / operator.

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

Performs the / operation.

impl<Shape> DivAssign<f64> for Tensor<Shape>

where Shape: TensorShape,

fn div_assign(&mut self, rhs: Float)

Performs the /= operation.

impl<const N: usize> From<[f64; N]> for Tensor<Dim<N, Nil>>

fn from(value: [Float; N]) -> Self

Converts to this type from the input type.

impl<Shape> Mul<&Tensor<Shape>> for &Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the * operator.

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

Performs the * operation.

impl<Shape> Mul<&Tensor<Shape>> for Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the * operator.

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

Performs the * operation.

impl<Shape> Mul<f64> for Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the * operator.

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

Performs the * operation.

impl<Shape> MulAssign<&Tensor<Shape>> for Tensor<Shape>

where Shape: TensorShape,

fn mul_assign(&mut self, rhs: &Tensor<Shape>)

Performs the *= operation.

impl<Shape> MulAssign<f64> for Tensor<Shape>

where Shape: TensorShape,

fn mul_assign(&mut self, rhs: Float)

Performs the *= operation.

impl<Shape> Sub<&Tensor<Shape>> for &Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the - operator.

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

Performs the - operation.

impl<Shape> Sub<&Tensor<Shape>> for Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the - operator.

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

Performs the - operation.

impl<Shape> Sub<f64> for Tensor<Shape>

where Shape: TensorShape,

type Output = Tensor<Shape>

The resulting type after applying the - operator.

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

Performs the - operation.

impl<Shape> SubAssign<&Tensor<Shape>> for Tensor<Shape>

where Shape: TensorShape,

fn sub_assign(&mut self, rhs: &Tensor<Shape>)

Performs the -= operation.

impl<Shape> SubAssign<f64> for Tensor<Shape>

where Shape: TensorShape,

fn sub_assign(&mut self, rhs: Float)

Performs the -= operation.