Enum Tensor 

pub enum Tensor {
    D1(Vec<f32>),
    D3(Vec<Vec<Vec<f32>>>),
}

Multi-dimensional tensor representation for intermediate activations

Variants

D1(Vec<f32>)

1D tensor [size]

D3(Vec<Vec<Vec<f32>>>)

3D tensor [channels, height, width]

Implementations

impl Tensor

pub fn as_1d(&self) -> Result<&Vec<f32>>

Extract as 1D tensor

pub fn as_3d(&self) -> Result<&Vec<Vec<Vec<f32>>>>

Extract as 3D tensor

pub fn size(&self) -> usize

Get the total number of elements

Trait Implementations

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.