Struct l2::tensor::Tensor

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

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<&'a Tensor<'a>> 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<&'a Tensor<'a>> 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<&'a Tensor<'a>> 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<Tensor<'a>> for Tensor<'a>

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

This method tests for self and other values to be equal, and is used by ==.

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

This method tests for !=.

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

impl<'a> Sub<&'a Tensor<'a>> 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.