Struct Tensor

pub struct Tensor<'a> {
    pub data: Vec<f32>,
    pub shape: Vec<usize>,
    pub strides: Vec<usize>,
    /* private fields */
}

Fields

data: Vec<f32>

shape: Vec<usize>

strides: Vec<usize>

Implementations

impl<'a> Tensor<'a>

pub fn new<'b>( data: Vec<f32>, shape: &[usize], ) -> Result<Tensor<'b>, TensorError>

pub fn new_no_grad<'b>( data: Vec<f32>, shape: &[usize], ) -> Result<Tensor<'b>, TensorError>

pub fn zeros<'b>(shape: &[usize]) -> Result<Tensor<'b>, TensorError>

pub fn normal<'b>( shape: &[usize], mean: f32, std: f32, ) -> Result<Tensor<'b>, TensorError>

pub fn uniform<'b>( shape: &[usize], low: f32, high: f32, ) -> Result<Tensor<'b>, TensorError>

pub fn slice( &'a self, logical_indices: &[[isize; 2]], ) -> Result<Self, TensorError>

pub fn view(&'a self, shape: &[isize]) -> Result<Self, TensorError>

pub fn pow(&self, exp: f32) -> Result<Tensor<'_>, TensorError>

pub fn sqrt(&self) -> Result<Tensor<'_>, TensorError>

pub fn exp(&self) -> Result<Tensor<'_>, TensorError>

pub fn log10(&self) -> Result<Tensor<'_>, TensorError>

pub fn log(&self) -> Result<Tensor<'_>, TensorError>

pub fn abs(&self) -> Result<Tensor<'_>, TensorError>

pub fn sin(&self) -> Result<Tensor<'_>, TensorError>

pub fn cos(&self) -> Result<Tensor<'_>, TensorError>

pub fn tan(&self) -> Result<Tensor<'_>, TensorError>

pub fn sum(&self, dim: isize) -> Result<Tensor<'_>, TensorError>

pub fn mean(&self, dim: isize) -> Result<Tensor<'_>, TensorError>

pub fn max(&self, dim: isize) -> Result<Tensor<'_>, TensorError>

pub fn min(&self, dim: isize) -> Result<Tensor<'_>, TensorError>

pub fn argmax(&self, dim: isize) -> Result<Tensor<'_>, TensorError>

pub fn argmin(&self, dim: isize) -> Result<Tensor<'_>, TensorError>

pub fn matmul(&'a self, rhs: &'a Tensor<'_>) -> Result<Tensor<'_>, TensorError>

pub fn concat( &self, rhs: &'a Tensor<'_>, dim: isize, ) -> Result<Tensor<'_>, TensorError>

pub fn transpose(&self) -> Result<Tensor<'_>, TensorError>

pub fn clone_no_grad(&self) -> Result<Tensor<'_>, TensorError>

pub fn backward(&self)

pub fn clear(&self)

Trait Implementations

impl<'a> Add for &'a Tensor<'a>

type Output = Tensor<'a>

The resulting type after applying the + operator.

fn add(self, other: &'a Tensor<'_>) -> Tensor<'a>

Performs the + operation.

impl<'a> Clone for Tensor<'a>

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for Tensor<'a>

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

Formats the value using the given formatter.

impl<'a> Display for Tensor<'a>

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

Formats the value using the given formatter.

impl<'a> Div for &'a Tensor<'a>

type Output = Tensor<'a>

The resulting type after applying the / operator.

fn div(self, other: &'a Tensor<'_>) -> Tensor<'a>

Performs the / operation.

impl<'a> Mul for &'a Tensor<'a>

type Output = Tensor<'a>

The resulting type after applying the * operator.

fn mul(self, other: &'a Tensor<'_>) -> Tensor<'a>

Performs the * operation.

impl<'a> PartialEq for Tensor<'a>

fn eq(&self, other: &Tensor<'a>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<'a> Sub for &'a Tensor<'a>

type Output = Tensor<'a>

The resulting type after applying the - operator.

fn sub(self, other: &'a Tensor<'_>) -> Tensor<'a>

Performs the - operation.

impl<'a> StructuralPartialEq for Tensor<'a>