Struct Tensor 

pub struct Tensor { /* private fields */ }

Implementations

impl Tensor

pub fn expand(&self, new_shape: &[usize]) -> Tensor

impl Tensor

pub fn new(data: &[f32], shape: &[usize]) -> Self

pub fn new_with_storage(storage: Storage, shape: &[usize]) -> Self

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

pub fn full(shape: &[usize], value: f32) -> Self

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

pub fn storage(&self) -> &Storage

pub fn to_wgpu(&self) -> Self

pub fn to_cpu(&self) -> Self

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

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

pub fn set_requires_grad(self, requires_grad: bool) -> Self

pub fn set_requires_grad_mut(&mut self, requires_grad: bool)

pub fn requires_grad(&self) -> bool

pub fn data(&self) -> RwLockReadGuard<'_, Vec<f32>>

pub fn data_mut(&self) -> RwLockWriteGuard<'_, Vec<f32>>

pub fn grad(&self) -> Option<Tensor>

pub fn zero_grad(&self)

pub fn accumulate_grad(&self, grad: &Tensor)

pub fn backward(&self)

pub fn backward_step(&self)

pub fn detach(&self) -> Tensor

Returns a new Tensor, detached from the current graph. The result will never require gradient.

pub fn set_op(&mut self, op: Arc<dyn BackwardOp>)

pub fn matmul(&self, rhs: &Tensor) -> Tensor

pub fn t(&self) -> Tensor

pub fn sub(&self, rhs: &Tensor) -> Tensor

pub fn add(&self, rhs: &Tensor) -> Tensor

pub fn neg(&self) -> Tensor

pub fn relu(&self) -> Tensor

pub fn sigmoid(&self) -> Tensor

pub fn tanh(&self) -> Tensor

pub fn softmax(&self, dim: i64) -> Tensor

pub fn conv2d( &self, weight: &Tensor, stride: (usize, usize), padding: (usize, usize), ) -> Tensor

pub fn max_pool2d( &self, kernel_size: (usize, usize), stride: (usize, usize), padding: (usize, usize), ) -> Tensor

pub fn batch_norm2d( &self, gamma: Option<&Tensor>, beta: Option<&Tensor>, running_mean: &Tensor, running_var: &Tensor, training: bool, momentum: f32, eps: f32, ) -> Tensor

pub fn layer_norm( &self, normalized_shape: &[usize], weight: Option<&Tensor>, bias: Option<&Tensor>, eps: f32, ) -> Tensor

pub fn permute(&self, dims: &[usize]) -> Tensor

pub fn transpose(&self, dim0: usize, dim1: usize) -> Tensor

pub fn contiguous(&self) -> Tensor

pub fn is_contiguous(&self) -> bool

pub fn normal_(&self, mean: f32, std: f32)

pub fn uniform_(&self, low: f32, high: f32)

pub fn fill_(&self, value: f32)

pub fn reshape(&self, new_shape: &[usize]) -> Tensor

pub fn mul(&self, rhs: &Tensor) -> Tensor

pub fn div(&self, rhs: &Tensor) -> Tensor

pub fn matmul_relu(&self, rhs: &Tensor) -> Tensor

pub fn sgd_step(&self, grad: &Tensor, lr: f32) -> Tensor

pub fn copy_(&self, src: &Tensor)

pub fn copy_from_slice(&self, src: &[f32])

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 Add for Tensor

type Output = Tensor

The resulting type after applying the + operator.

fn add(self, rhs: Tensor) -> 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 for &Tensor

type Output = Tensor

The resulting type after applying the / operator.

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

Performs the / operation.

impl Div for Tensor

type Output = Tensor

The resulting type after applying the / operator.

fn div(self, rhs: Tensor) -> Tensor

Performs the / operation.

impl Mul for &Tensor

type Output = Tensor

The resulting type after applying the * operator.

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

Performs the * operation.

impl Mul for Tensor

type Output = Tensor

The resulting type after applying the * operator.

fn mul(self, rhs: Tensor) -> Tensor

Performs the * operation.

impl PartialEq for Tensor

fn eq(&self, other: &Self) -> 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 Sub for &Tensor

type Output = Tensor

The resulting type after applying the - operator.

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

Performs the - operation.

impl Sub for Tensor

type Output = Tensor

The resulting type after applying the - operator.

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

Performs the - operation.