Struct Tensor 

pub struct Tensor { /* private fields */ }

Implementations

impl Tensor

pub fn zeros(shape: &[usize]) -> Self

pub fn ones(shape: &[usize]) -> Self

pub fn from_vec(shape: &[usize], data: Vec<f64>) -> Self

pub fn from_fn(shape: &[usize], f: impl Fn(&[usize]) -> f64) -> Self

pub fn scalar(val: f64) -> Self

pub fn identity(n: usize) -> Self

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

pub fn rank(&self) -> usize

pub fn size(&self) -> usize

pub fn data(&self) -> &[f64]

pub fn get(&self, indices: &[usize]) -> f64

pub fn set(&mut self, indices: &[usize], value: f64)

pub fn scale(&self, s: f64) -> Self

pub fn map(&self, f: impl Fn(f64) -> f64) -> Self

pub fn elementwise(&self, other: &Tensor, f: impl Fn(f64, f64) -> f64) -> Self

pub fn sum(&self) -> f64

pub fn max(&self) -> f64

pub fn min(&self) -> f64

pub fn frobenius_norm(&self) -> f64

Trait Implementations

impl Add for &Tensor

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: Self) -> Tensor

Performs the + operation.

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate 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 Div<f64> for &Tensor

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, s: f64) -> Tensor

Performs the / operation.

impl Mul<f64> for &Tensor

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, s: f64) -> Tensor

Performs the * operation.

impl Neg for &Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn neg(self) -> Tensor

Performs the unary - operation.

impl Sub for &Tensor

type Output = Tensor

The resulting type after applying the - operator.

fn sub(self, rhs: Self) -> Tensor

Performs the - operation.