Struct DenseTensor 

pub struct DenseTensor {
    pub data: Vec<f64>,
    pub shape: Vec<usize>,
}

Dense tensor for input/output

Fields

data: Vec<f64>

Tensor data in row-major order

shape: Vec<usize>

Shape of the tensor

Implementations

impl DenseTensor

pub fn new(data: Vec<f64>, shape: Vec<usize>) -> Self

Create tensor from data and shape

pub fn zeros(shape: Vec<usize>) -> Self

Create zeros tensor

pub fn ones(shape: Vec<usize>) -> Self

Create ones tensor

pub fn random(shape: Vec<usize>, seed: u64) -> Self

Create random tensor

pub fn order(&self) -> usize

Get tensor order (number of dimensions)

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

Get linear index from multi-index

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

Get element at multi-index

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

Set element at multi-index

pub fn frobenius_norm(&self) -> f64

Compute Frobenius norm

pub fn reshape(&self, new_shape: Vec<usize>) -> Self

Reshape tensor (view only, same data)

Trait Implementations

impl Clone for DenseTensor

fn clone(&self) -> DenseTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DenseTensor

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

Formats the value using the given formatter.