Struct Tensor

pub struct Tensor { /* private fields */ }

Implementations

impl Tensor

pub fn new(tensor: TensorStorage, requires_grad: bool) -> Self

pub fn tensor(&self) -> &TensorStorage

pub fn tensor_mut(&mut self) -> &mut TensorStorage

pub fn requires_grad(&self) -> &bool

pub fn grad_fn(&self) -> Option<Rc<dyn GradientFunction>>

pub fn set_grad_fn(&mut self, grad_fn: Option<Rc<dyn GradientFunction>>)

pub fn grad(&self) -> Option<Rc<RefCell<TensorStorage>>>

pub fn backward(&mut self)

Trait Implementations

impl Add<&Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the + operator.

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

Performs the + operation.

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a copy 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 Mul<&Tensor> for &Tensor

type Output = Tensor

The resulting type after applying the * operator.

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

Performs the * operation.

impl TensorCreation for Tensor

fn zeros(shape: Vec<usize>, requires_grad: Option<bool>) -> Self

fn ones(shape: Vec<usize>, requires_grad: Option<bool>) -> Self

fn from_ndarray<S, D, T>( data: &ArrayBase<S, D>, requires_grad: Option<bool>, ) -> Self

where S: Data<Elem = T>, T: AsPrimitive<f32>, D: Dimension,

impl TensorOps for Tensor

fn add_tensor(&self, other: &Self) -> Self

fn mul_tensor(&self, other: &Self) -> Self

fn sum(&self) -> Self

fn mean(&self) -> Self

fn product(&self) -> Self

fn sub_tensor(&self, other: &Self) -> Self

fn div_tensor(&self, other: &Self) -> Self

fn add_f32(&self, other: f32) -> Self

fn sub_f32(&self, other: f32) -> Self

fn mul_f32(&self, other: f32) -> Self

fn div_f32(&self, other: f32) -> Self

fn add_tensor_assign(&mut self, other: &Self)

fn sub_tensor_assign(&mut self, other: &Self)

fn mul_tensor_assign(&mut self, other: &Self)

fn div_tensor_assign(&mut self, other: &Self)

fn add_f32_assign(&mut self, other: f32)

fn sub_f32_assign(&mut self, other: f32)

fn mul_f32_assign(&mut self, other: f32)

fn div_f32_assign(&mut self, other: f32)

fn matmul(&self, other: &Self) -> Self

impl TensorShape for Tensor

fn reshape(&mut self, new_shape: Vec<usize>)

fn transpose(&mut self)

fn permute(&mut self, dims: &[usize])

fn flatten(&mut self)

fn squeeze(&mut self)

fn unsqueeze(&mut self, dim: usize)

fn broadcast(&self, new_shape: &[usize]) -> Self

fn compute_broadcast_shape(&self, target_shape: &[usize]) -> Vec<usize>

fn compute_broadcast_strides(&self, broadcast_shape: &[usize]) -> Vec<usize>

fn pad_shape(&self, target_rank: usize) -> Vec<usize>

fn broadcast_tensors(a: &Self, b: &Self) -> (Self, Self)